PyPI page
Home page
Author:
None
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.12
Required dependencies:
absl-py
|
einshape
|
immutabledict
|
jax
|
jaxlib
|
jaxtyping
|
pydantic
|
qwix
|
tensorboardx
|
tqdm
|
typeguard
|
typing_extensions
Optional dependencies:
chex
|
flatbuffers
|
flax
|
google-benchmark
|
hypothesis
|
jax
|
libtpu
|
nvidia-cudnn-cu12
|
pytest
|
pytest-xdist
|
xprof
Downloads last day:
5,336
Downloads last week:
66,323
Downloads last month:
225,567