PyPI page
Author:
Ani Udipi
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
A Pallas Custom Kernel Library.
Latest version:
0.0.16
Required dependencies:
absl-py
|
einshape
|
immutabledict
|
jax
|
jaxlib
|
jaxtyping
|
pydantic
|
qwix
|
tensorboardx
|
tqdm
|
xprof
Optional dependencies:
chex
|
cuequivariance-jax
|
cuequivariance-ops-jax-cu13
|
flatbuffers
|
flax
|
google-benchmark
|
hypothesis
|
jax
|
jax-triton
|
nvidia-cudnn-cu13
|
nvidia-cudnn-frontend
|
pytest
|
pytest-xdist
|
triton
Downloads last day:
56,281
Downloads last week:
216,645
Downloads last month:
491,393
7-day smoothing