PyPI page
Home page
Author:
Igor Sokolov
License:
Apache-2.0
Summary:
Differentiable eigenvalue decomposition with JAX — CUDA 12.8 (GPU) build
Latest version:
0.4.1
Required dependencies:
jax
|
jaxlib
|
numpy
|
nvidia-cublas-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nvjitlink-cu12
Optional dependencies:
pytest
|
scipy
Downloads last day:
1
Downloads last week:
30
Downloads last month:
217