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.79
Required dependencies:
beartype
|
chex
|
einops
|
jax
|
jaxlib
|
jaxtyping
|
pydantic
|
tqdm
Optional dependencies:
eformer
|
jax
|
jax-tvm-ffi
|
nvidia-cutlass-dsl
|
tb-nightly
|
triton
|
xprof
|
xprof-nightly
Downloads last day:
435
Downloads last week:
2,324
Downloads last month:
14,240