Aria is Your AI Research Assistant Powered by GPT Large Language Models

UNet diffusion model in pure CUDA

Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"

by ex-googlers, for ex-googlers - a lookup table of similar tech & services

seqax = sequence modeling + JAX

PyTorch code and models for V-JEPA self-supervised learning from video.

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

A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.

A JAX research toolkit for building, editing, and visualizing neural networks.

Grok open release

A Production-ready Reinforcement Learning AI Agent Library brought by the Applied Reinforcement Learning team at Meta.

Flax is a neural network library for JAX that is designed for flexibility.

Python Implementation of Reinforcement Learning: An Introduction

Monte Carlo tree search in JAX

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

Really Fast End-to-End Jax RL Implementations

Multi-Agent Reinforcement Learning with JAX

A simple, performant and scalable Jax LLM!

A high-throughput and memory-efficient inference and serving engine for LLMs

AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

Kandinsky 2 — multilingual text2image latent diffusion model

Convert natural language to LaTeX within Overleaf using LLMs

GLM-130B: An Open Bilingual Pre-Trained Model (ICLR 2023)

Task-based datasets, preprocessing, and evaluation for sequence models.

Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"