PyPI page
Home page
Author:
Erfan Zare Chavoshi
License:
Apache-2.0
Summary:
Accelerate, Optimize performance with streamlined training and serving options with JAX.
Latest version:
0.0.80.3
Required dependencies:
beartype
|
chex
|
einops
|
jax
|
jaxlib
|
jaxtyping
|
pydantic
|
tqdm
Optional dependencies:
apache-tvm-ffi
|
eformer
|
jax
|
jax-cuda13-pjrt
|
jax-cuda13-plugin
|
jax-tvm-ffi
|
nvidia-cudnn-cu12
|
nvidia-cutlass-dsl
|
nvidia-cutlass-dsl-libs-base
|
nvidia-cutlass-dsl-libs-cu13
|
tb-nightly
|
tilelang
|
triton
|
xprof
|
xprof-nightly
Downloads last day:
4
Downloads last week:
392
Downloads last month:
4,402