PyPI
avitai/substrax
JAX/Flax NNX training infrastructure: device meshes and SPMD sharding, an Orbax checkpoint store, early stopping and callbacks, W&B and MLflow logging.
MIT26 sept 2026 · métricas 2.10.0
Todas las etiquetas
Etiqueta del catálogo
#tensor-parallel
Todos los repositorios del registro público que llevan esta etiqueta, procedente de sus topics de GitHub o de las palabras clave que publican sus registros de paquetes. La salud se mide con la misma metodología versionada que el resto del registro.
1 registro