Module layout 

Memory Layout Optimizer

Automatically chooses the best memory layout for each operation This gives us the final edge over JAX!

Structs

DeviceInfo

Device capabilities

LayoutOptimizer

Layout optimizer

Enums

MemoryLayout

Memory layout formats

OperationType

Operation types for layout selection