helmholtz-analytics / heat Star 194 Code Issues Pull requests Discussions Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python python data-science machine-learning hpc gpu numpy mpi pytorch distributed parallelism data-analytics tensors data-processing multi-gpu mpi4py massive-datasets multi-node-cluster array-api Updated May 21, 2024 Python
pyxu-org / pyxu Star 112 Code Issues Pull requests Modular and scalable computational imaging in Python with GPU/out-of-core computing. machine-learning modular scalable computer-vision image-reconstruction python-library image-processing artificial-intelligence bayesian-methods computational-imaging image-restoration image-enhancement sensing matrix-free sparse-learning inverse-problem foundation-models array-api python-hpc array-computing Updated May 7, 2024 Python
sdpython / onnx-array-api Star 7 Code Issues Pull requests Array APIs to write ONNX Graphs machine-learning numpy onnx array-api Updated Apr 18, 2024 Python
ar90n / vanilla-roll Star 3 Code Issues Pull requests array-api based simple volume renderer python dicom medical-imaging volume-rendering nifti medical-images mha my-projects array-api Updated May 22, 2024 Python
GalacticDynamics / quaxed Star 2 Code Issues Pull requests Pre-`quaxify`d libraries for use in multiple dispatch over abstract array types in JAX. array multiple-dispatch ducktyping jax array-api Updated May 20, 2024 Python