TORAX: Tokamak transport simulation in JAX

A library of surrogate transport models for tokamak fusion.

Leetcode for Pytorch

A gui for the computer algebra system Maxima built with wxWidgets

The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬

Darwin Gödel Machine: Open-Ended Evolution of Self-Improving Agents

Open-source implementation of AlphaEvolve

The best ChatGPT that $100 can buy.

Python toolbox for optimization on Riemannian manifolds with support for automatic differentiation

kernels, of the mega variety

Mirage Persistent Kernel: Compiling LLMs into a MegaKernel

Generative Models by Stability AI

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Python SDK, Proxy Server (AI Gateway) to call 100+ LLM APIs in OpenAI (or native) format, with cost tracking, guardrails, loadbalancing and logging. [Bedrock, Azure, OpenAI, VertexAI, Cohere, Anthr…

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

A minimal GPU design in Verilog to learn how GPUs work from the ground up

A TTS model capable of generating ultra-realistic dialogue in one pass.

Tokamax: A GPU and TPU kernel library.

Chai-1, SOTA model for biomolecular structure prediction

Official repository for the Boltz biomolecular interaction models

A JAX-native LLM Post-Training Library

Train transformer language models with reinforcement learning.

verl: Volcano Engine Reinforcement Learning for LLMs

Repository for code used in the xVal paper