Downloads in the 30 and 7 days to October 10, 2026. Refreshed weekly.
Package Information
No reported malware or high/critical advisories · Details
The current version has no OSV.dev malware reports and no critical or high-severity vulnerabilities.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100More insights and metrics
Downloads by OS
Python Versions
Top Countries
Weekly Downloads
GitHub Stars
Dependencies
- beartype >=0.22.2
- chex >=0.1.91
- einops >=0.8.1
- jax ~=0.10.0
- jaxlib ~=0.10.0
- jaxtyping >=0.3.2
- pydantic >=2.11.10
- tqdm >=4.67.1
18 optional dependencies
-
apache-tvm-ffi==0.1.8.post2[cuda]extra == "cuda"
-
apache-tvm-ffi==0.1.8.post2[tilelang]extra == "tilelang"
-
eformer[dev]extra == "dev"
-
jax~=0.10.0[cuda]extra == "cuda"
-
jax~=0.10.0[tpu]extra == "tpu"
-
jax-cuda13-pjrt~=0.10.0[cuda]extra == "cuda"
-
jax-cuda13-plugin~=0.10.0[cuda]extra == "cuda"
-
jax-tvm-ffi==0.1.2[cuda]extra == "cuda"
-
nvidia-cudnn-cu12==9.12.0.46[tilelang]extra == "tilelang"
-
nvidia-cutlass-dsl==4.5.2[cuda]extra == "cuda"
-
nvidia-cutlass-dsl-libs-base==4.5.2[cuda]extra == "cuda"
-
nvidia-cutlass-dsl-libs-cu13==4.5.2[cuda]extra == "cuda"
-
tb-nightly>=2.21.0a20250820[profile]extra == "profile"
-
tilelang==0.1.9[tilelang]extra == "tilelang"
-
triton==3.6.0[cuda]extra == "cuda"
-
triton==3.6.0[triton]extra == "triton"
-
xprof>=2.20.6[profile]extra == "profile"
-
xprof-nightly>=2.21.6a20250820[profile]extra == "profile"
Used By
- easydel 1.8K/mo