Fast, flexible and easy to use probabilistic modelling in Python
Medical imaging toolkit for deep learning
AIMET is a library that provides advanced quantization and compression
Fast and accurate AI powered file content types detection
State-of-the-art diffusion models for image and audio generation
Training framework for Stable Baselines3 reinforcement learning agents
Monte Carlo tree search in JAX
The Library for LLM-based multi-agent applications
Build high-performance AI models with modular building blocks
A TensorFlow library for applied reinforcement learning
ChatArena (or Chat Arena) is a Multi-Agent Language Game Environments
An open-source toolkit for monitoring Language Learning Models (LLMs)
A refreshing functional take on deep learning
Advanced techniques for RAG systems
Deep learning optimization library: makes distributed training easy
The NVIDIA AgentIQ toolkit is an open-source library
Simple, unified interface to multiple Generative AI providers
The largest collection of PyTorch image encoders / backbones
Advanced evolutionary computation library built on top of PyTorch
LLM abstractions that aren't obstructions
⚡ Building applications with LLMs through composability ⚡
A library for accelerating Transformer models on NVIDIA GPUs
A batteries-included library for building AI-powered software
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch