#pytorch

Simple, safe way to store and distribute tensors

Graph Neural Network Library for PyTorch

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools, with 10x faster training through evolutionary hyperparameter optimization.

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization pipelines, and OpenAI-compatible/MCP serving.

A highly efficient implementation of Gaussian Processes in PyTorch

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultralytics / MMEngine / Keras etc.

Data manipulation and transformation for audio signal processing, powered by PyTorch

Brevitas: neural network quantization in PyTorch

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

High-performance automatic differentiation of LLVM and MLIR.

The fastai deep learning library

cuDNN Frontend is NVIDIA's modern, open-source entry point to the cuDNN library and a growing collection of high-performance open-source kernels.

Support PyTorch model conversion with LiteRT.

A Python library for anomaly detection across tabular, time series, graph, text, image, and audio data. 60+ detectors, benchmark-backed ADEngine orchestration, and an agentic workflow for AI agents.

Soup turns the pain of LLM fine-tuning into a simple workflow. One config, one command, done.

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Compatibility layer for common array libraries to support the Array API

High-performance safetensors model loader

An open source implementation of CLIP.

Tensor Fusion is a state-of-the-art GPU virtualization and pooling solution designed to optimize GPU cluster utilization to its fullest potential.

torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.

Visualizer for neural network, deep learning and machine learning models

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.

Tools for computational pathology

Tensor Fusion is a state-of-the-art GPU virtualization and pooling solution designed to optimize GPU cluster utilization to its fullest potential.