Build a machine learning model from a prompt
Official inference library for Mistral models
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
A library for accelerating Transformer models on NVIDIA GPUs
Training PyTorch models with differential privacy
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
Graph Neural Network Library for PyTorch
Official python implementation of UTCP. UTCP is an open standard
HunyuanVideo: A Systematic Framework For Large Video Generation Model
Uncover insights, surface problems, monitor, and fine tune your LLM
A Repo For Document AI
Agents write python code to call tools and orchestrate other agents
Tools for merging pretrained large language models
An API standard for multi-agent reinforcement learning environments
Library for serving Transformers models on Amazon SageMaker
Our first fully AI generated deep learning system
Foundation Models for Time Series
tiktoken is a fast BPE tokeniser for use with OpenAI's models
Evaluation and Tracking for LLM Experiments
Fast, flexible and easy to use probabilistic modelling in Python
Agent-ready RPA suite with visual workflow automation tools engine
A unified framework for scalable computing
Deep learning optimization library making distributed training easy
Training framework for Stable Baselines3 reinforcement learning agents