Search results

Add logical axis to mesh mapping in init

Status: Open (in progress).

AI-Hypercomputer/maxdiffusionnumber 263

#263 In AI-Hypercomputer/maxdiffusion;

· coolkp opened on Oct 9, 2025

Fix namedsharding

Status: Open (in progress).

AI-Hypercomputer/maxdiffusionnumber 190

#190 In AI-Hypercomputer/maxdiffusion;

· coolkp opened on Jun 25, 2025

Remove noop spmd_mode check, correct type annotations in attention flax

Status: Draft (not ready).

AI-Hypercomputer/maxdiffusionnumber 170

#170 In AI-Hypercomputer/maxdiffusion;

· coolkp opened on May 1, 2025