🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

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

JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf

Running large language models on a single GPU for throughput-oriented scenarios.

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement

Entropy Based Sampling and Parallel CoT Decoding

Implementation of RT1 (Robotic Transformer) in Pytorch