ShadowPyExpect is a framework for classical shadow computation, seamlessly integrating Python and C++ to perform high-performance expectation calculations.

Implementation of Autoregressive Diffusion in Pytorch

Implementation of rectified flow and some of its followup research / improvements in Pytorch

Implementing a ChatGPT-like LLM in PyTorch from scratch, step by step

A generative model for programmable protein design

Text-based modeling of materials.

LLM101n: Let's build a Storyteller

Flow-matching algorithms in JAX

Video+code lecture on building nanoGPT from scratch

TextGrad: Automatic ''Differentiation'' via Text -- using large language models to backpropagate textual gradients.

CodiumAI Cover-Agent: An AI-Powered Tool for Automated Test Generation and Code Coverage Enhancement! 💻🤖🧪🐞

Mixed continous/categorical flow-matching model for de novo molecule generation.

Model parallel transformers in JAX and Haiku

Jupyter notebook with Pytorch implementation of Neural Ordinary Differential Equations

Curated list of data science interview questions and answers

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

JMP is a Mixed Precision library for JAX.

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

variational free-energy of dense hydrogen

JAX - A curated list of resources https://github.com/google/jax

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

A code to generate atomic structure with symmetry

[GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-simple, user-friendly …

The Wren sits on its Roost in the Aviary.

Space Group Informed Transformer for Crystalline Materials Generation