For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo trait

WeightLoader

B/weight-side comptime policy for the shared Apple GEMM body.

Carries the B operand dtype (b_type), the MMA accumulator dtype (accum_type; fp32 for dense/fp8/fp4, int32 for a later int8 slice), and the SMEM-staging policy (needs_smem / smem_elems). The A loader still issues the MMA; this policy tells the body how to source and size the B side.

Implemented traits

AnyType

`comptime` members

`accum_type`

comptime accum_type

`b_type`

comptime b_type

`needs_smem`

comptime needs_smem

`smem_elems`

comptime smem_elems

Implemented traits​

comptime members​

accum_type​

b_type​

needs_smem​

smem_elems​