Making large AI models cheaper, faster and more accessible

You like pytorch? You like micrograd? You love tinygrad! ❤️

Make Your Company Data Driven. Connect to any data source, easily visualize, dashboard and share your data.

A generative world for general-purpose robotics & embodied AI learning.

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

💨 Fast, Async-ready, Openapi, type hints based framework for building APIs

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

Flagsmith is an open source feature flagging and remote config service. Self-host or use our hosted version at https://app.flagsmith.com.

Vector (and Scalar) Quantization, in Pytorch

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

A modular RL library to fine-tune language models to human preferences

The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”

Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch

Code repository for supporting the paper "Atlas Few-shot Learning with Retrieval Augmented Language Models",(https//arxiv.org/abs/2208.03299)

Python port of R's Comprehensive Dynamic Time Warp algorithms package

🐍 | Python library for RunPod API and serverless worker SDK.

A helper library to connect into Amazon SageMaker with AWS Systems Manager and SSH (Secure Shell)

unofficial implementation of the High Fidelity Neural Audio Compression

Scripts for computing the Intelligibility and CLVP scores for evaluating TTS models

Audiogen Codec

Fast, Modern, and Low Precision PyTorch Optimizers

A self-hosted HuggingFace alternative

📄Small Batch Size Training for Language Models

A python library that allow you to build full swiftUI apps using only python

Implementation for "SoundCLR: Contrastive Learning of Representations For Improved Environmental Sound Classification," in pytorch.

Official repo for the paper Thought Tokens - Letting language models 'think'