Mojo package

layout

Modules

element: Provides element-based access to memory using layout-driven vectorization.
int_tuple: Hierarchical integer tuple data structures for high-performance tensor operations.
layout: Provides a high-performance tensor layout system for memory mapping and indexing.
layout_tensor:
math:
nd_buffer_stub:
runtime_layout: Provides the RuntimeLayout type and functions for working with it. You can use RuntimeLayout to define a layout where the dimensions are not known at compile time.
runtime_tuple:
swizzle: Defines swizzle layouts for optimizing memory access patterns, particularly in shared memory for GPU kernels, to mitigate bank conflicts.
tensor_builder:
tensor_core: This module provides abstractions for using Tensor Cores do to arithmetic and matrix operations.
tensor_core_async: This module provides abstractions for using Async Tensor Cores to perform asynchronous matrix multiplication operations.
tma_async: Tensor Memory Accelerator (TMA) Asynchronous Operations Module