rdangovs

llama

Inference code for Llama models

llama2.c

Inference Llama 2 in one file of pure C

RWKV-LM

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference, saves VRAM, fast training, "infinite" ctx_len, and free sentence embedding.

h2ogpt

Private chat with local GPT with document, images, video, etc. 100% private, Apache 2.0. Supports oLLaMa, Mixtral, llama.cpp, and more. Demo: https://gpt.h2o.ai/ https://codellama.h2o.ai/

mistral-src

Reference implementation of Mistral AI 7B v0.1 model.

consistency_models

Official repo for consistency models.

graphcast

YaLM-100B

Pretrained language model with 100B parameters

s4

Structured state space sequence models

pythia

The hub for EleutherAI's work on interpretability and learning dynamics

prm800k

800,000 step-level correctness labels on LLM solutions to MATH problems

DiffDock

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking

LOMO

LOMO: LOw-Memory Optimization

e3nn

A modular framework for neural networks with Euclidean symmetry

nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

evosax

Evolution Strategies in JAX 🦎

mace

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.

flaxformer

agd

Automatic gradient descent

srbench

A living benchmark framework for symbolic regression

vistext

VisText is a benchmark dataset for semantically rich chart captioning.

rotational-unit-of-memory

RUM

Single-Player-MCTS

🌳 Python implementation of single-player Monte-Carlo Tree Search.

OccamNet_Public

wp-hungarian

contrastive-phase-transitions

Studying Phase Transitions in Contrastive Learning with Physics-Inspired Datasets

ModelStitching_LookingForFunctionalSimilarityBetweenRepresentations

Contrastive Learning SuperUROP Fall 2021 - Spring 2022

state-spaces

Sequence Modeling with Structured State Spaces

supervised-CL

srbench

A living benchmark framework for symbolic regression