NumPy & SciPy for GPU
- cuda
- cudnn
- cublas
- cusolver
54,641 visible repositories in the index
1–3 of 3 · Most stars
NumPy & SciPy for GPU
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
:dart: Accurate sums and dot products for Python.