Cuml alternatives for AI agents.

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

The world's simplest facial recognition api for Python and the command line

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Datasets, Transforms and Models specific to Computer Vision

NLTK Source

A hyperparameter optimization framework

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

A python library for user-friendly forecasting and anomaly detection on time series.

Seamlessly integrate LLMs into scikit-learn.

The fastai deep learning library

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

Deep Learning for humans

Open3D: A Modern Library for 3D Data Processing

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Open Machine Learning Compiler Framework

Run, manage, and scale AI workloads on any AI infrastructure. Use one system to access & manage all AI compute (Kubernetes, Slurm, 20+ clouds, on-prem).