Graph Neural Network Library for PyTorch

Ray is a unified framework for scaling AI and Python applications. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

pyrelational is a python active learning library for rapidly implementing active learning pipelines from data management, model development (and Bayesian approximation), to creating novel active le…

The fundamental package for scientific computing with Python.

Model interpretability and understanding for PyTorch

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Geometric Computer Vision Library for Spatial AI