Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- async-lru ~=2.0
- braceexpand >=0.1.7
- chex >=0.1.86
- dataclasses-json ~=0.6.4
- datasets <5.0,>=3.1.0
- deepdiff >=8.6.2
- draccus >=0.11.5
- einops
- equinox !=0.12.0,>=0.11.7
- filelock ~=3.13
- fsspec <2027,>=2024.2
- gcsfs <2027,>=2024.2
- humanfriendly ==10.0
- immutabledict
- jax <0.11,>=0.9.2
- jaxtyping >=0.2.34
- jinja2
- jmp >=0.0.3
- kitoken >=0.10.2
- lenses
- marin-fray ==0.99
- marin-haliax ==0.99
- marin-rigging ==0.99
- marin-zephyr ==0.99
- optax <0.2.7,>=0.1.9
- protobuf >=6
- pyarrow >=11.0.0
- pydantic <3
- pytimeparse >=1.1.8
- safetensors <0.7.0,>=0.4.2
- tblib <4.0.0,>=1.7.0
- tensorstore <0.1.82,>=0.1.73
- tokenizers >=0.15.2
- tqdm-loggable >=0.2
- transformers <5.0,>=4.57.1
- wandb >0.24.0
- zstandard >=0.18.0
20 optional dependencies
- fastapi[serve]
- google-api-python-client[gcp]
- google-auth[gcp]
- google-cloud-storage[gcp]
- google-cloud-storage-transfer[gcp]
- jax[tpu]
- jax[gpu]
- jaxlib[tpu]
- kitoken[kitoken]
- libtpu[tpu]
- nvidia-cublas[gpu]
- nvidia-nccl-cu13[gpu]
- openai[serve]
- peft[torch-test]
- tensorboard[profiling]
- tensorboardx[profiling]
- tokamax[kernels]
- torch[torch-test]
- uvicorn[serve]
- xprof[profiling]