PyPI page
Home page
Author:
None
Summary:
JAX library for optimization and export of models for use with the UZU inference engine.
Latest version:
0.13.1
Required dependencies:
beartype
|
cattrs
|
click
|
einops
|
equinox
|
huggingface-hub
|
jax
|
jaxtyping
|
jinja2
|
ml-dtypes
|
nvidia-cutlass-dsl
|
optax
|
orbax-checkpoint
|
polars
|
requests
|
rich
|
soundfile
|
thefuzz
|
tokamax
|
tokenizers
|
typer
|
xxhash
Optional dependencies:
blobfile
|
fastapi
|
jax
|
pyaudio
|
tiktoken
|
torch
|
transformers
|
uvicorn
Downloads last day:
4
Downloads last week:
24
Downloads last month:
276