Factual package intelligence from PyPI

tax-dp 0.1.1 package intelligence

A fully sharded data parallel trainer for jax

pip install tax-dp

Current version
0.1.1
Python requirement
>=3.10
License
MIT
Distribution type
Pure Python wheel
Release files
2 (1 wheels, 1 source)
Download size
89.3 KiB
Release history
1 releases with files
Median release cadence
Not enough dated releases

tax-dp dependencies

PyPI declares 12 unique dependency rules for this release. Environment markers are shown when supplied by the project.

  1. jax ==0.4.31
  2. tqdm ==4.66.2
  3. wandb ==0.17.2
  4. black >=24.8.0
  5. flax ==0.8.4
  6. orbax-checkpoint ==0.6.4
  7. matplotlib >=3.9.1
  8. pip >=24.2
  9. jaxtyping >=0.2.34
  10. torch >=2.4.1
  11. transformers >=4.45.1
  12. datasets >=2.19.2

Compatibility and release files

tax-dp publishes 1 wheel and 1 source archive for version 0.1.1. Wheel platform tags: any.

No version-specific Python classifiers are declared.

Metadata completeness

PyPI does not currently declare: project URL. PyDeps marks these fields as unknown instead of guessing values.

Release activity and sources

PyPI lists 1 release with files. The first dated release is ; 1 release falls within the 365 days preceding the latest dated release. The current release files were uploaded on .

The project does not publish external project URLs in its PyPI metadata.