substrax 0.1.17

T1Covered by 1 source · first reported by PyPI / crates.io / RubyGems / Go index / NuGet at 25 Sept, 21:21 UTC

JAX/Flax NNX training infrastructure: device meshes and SPMD sharding, an Orbax checkpoint store, early stopping and callbacks, W&B and MLflow logging, and job runs on Modal, SkyPilot or locally