PyPI page
Home page
Author:
Jacob Tutt, Roeland Wiersema
Summary:
JAXMg provides a jittable C++/CUDA interface between JAX and NVIDIA cuSOLVERMp.
Latest version:
1.4.1
Required dependencies:
jax
Optional dependencies:
hippogriffe
|
jax
|
matplotlib
|
mike
|
mkdocs
|
mkdocs-exclude
|
mkdocs-material
|
mkdocstrings
|
nvidia-cusolvermp-cu12
|
nvidia-cusolvermp-cu13
|
pytest
Downloads last day:
29
Downloads last week:
151
Downloads last month:
1,726
7-day smoothing