Faster and easier training and deployments
Learning to Reason with Search for LLMs via Reinforcement Learning
A simple, performant and scalable Jax LLM
Train multi-step agents for real-world tasks using GRPO
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
Deep learning optimization library: makes distributed training easy
Integrate, train and manage any AI models and APIs with your database
The machine learning toolkit for time series analysis in Python
An open-source Chinese font derived from Fontworks' Klee One
DataDreamer: Prompt. Generate Synthetic Data. Train & Align Models
Implementation of Imagen, Google's Text-to-Image Neural Network
Retrieval and Retrieval-augmented LLMs
Toolkit for conversational AI
TextWorld is a sandbox learning environment for the training
Learning What You Want to Learn Using Programmable Gradient Info
Implementation and experiments of graph embedding algorithms
Training framework for Stable Baselines3 reinforcement learning agents
A Python library for learning and evaluating knowledge graph embedding
Powering Amazon custom machine learning chips
A deep matching model library for recommendations & advertising
The largest collection of PyTorch image encoders / backbones
Standalone, small, language-neutral
Python framework for adversarial attacks, and data augmentation
Implementation of Denoising Diffusion Probabilistic Model in Pytorch