Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
-
Updated
Nov 18, 2024 - Python
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
Modular and scalable computational imaging in Python with GPU/out-of-core computing.
array-api based simple volume renderer
Pre-quaxed libraries for use in multiple dispatch over abstract array types in JAX.
Add a description, image, and links to the array-api topic page so that developers can more easily learn about it.
To associate your repository with the array-api topic, visit your repo's landing page and select "manage topics."