Train models with JAX and Flax.
Language models, diffusion and JEPA, trained by one trainer on one device or a mesh.
uv pip install "dew-ml @ git+https://github.com/AshishKumar4/dew"
DewDewLanguage models, diffusion and JEPA, trained by one trainer on one device or a mesh.
uv pip install "dew-ml @ git+https://github.com/AshishKumar4/dew"