augmax

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Deep Learning for humans

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

The fastai deep learning library

? A ranked list of awesome machine learning Python libraries. Updated weekly.

Convert Machine Learning Code Between Frameworks

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Open3D: A Modern Library for 3D Data Processing

NumPy & SciPy for GPU