vwxyzjn

grok-1

Grok open release

openui

OpenUI let's you describe UI using your imagination, then see it rendered live.

mlx

MLX: An array framework for Apple silicon

uv

An extremely fast Python package installer and resolver, written in Rust.

mamba

cv

Print-friendly, minimalist CV page

minbpe

Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.

PokemonRedExperiments

Playing Pokemon Red with Reinforcement Learning

deskhop

Fast Desktop Switching Device

gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

OLMo

Modeling, training, eval, and inference code for OLMo

direct-preference-optimization

Reference implementation for DPO (Direct Preference Optimization)

summarize-from-feedback

Code for "Learning to summarize from human feedback"

axlearn

An Extensible Deep Learning Library

nanotron

Minimalistic large language model 3D-parallelism training

huak

My experimental Python package manager.

RingAttention

Transformers with Arbitrarily Large Context

code-eval

Run evaluation on LLMs using human-eval benchmark

text-clustering

Easily embed, cluster and semantically label text datasets

docstring_parser

Parse Python docstrings in various flavors.

pretraining-with-human-feedback

Code accompanying the paper Pretraining Language Models with Human Preferences

flashbax

⚡ Flashbax: Accelerated Replay Buffers in JAX

OSRL

🤖 Elegant implementations of offline safe RL algorithms in PyTorch

seqax

seqax = sequence modeling + JAX

oaib

Use the OpenAI Batch tool to make async batch requests to the OpenAI API.

jaxued

sebulba

🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX

PDFwriter

An OSX print to pdf-file printer driver

cogment-lab

A toolkit for practical Human-AI cooperation research

putting-dune