Stars
Claude Code skill + MCP server for managing Google Cloud TPU capacity and Gemma 4 vLLM serving on TPU VMs
Run ML workloads seamlessly on cloud TPUs and GPUs with a single Python decorator. No infrastructure management required.
TPU Stack Demo for Google IO 2026
Web app for crossword construction: designing a grid, filling the grid, creating straight or cryptic clues, and downloading as Exolve or .puz. All data is saved locally, no crosswords are sent to a…
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Implementing scalable LLMs in pure JAX (no third-party libraries)
The simplest, fastest repository for training/finetuning medium-sized GPTs.
A simple, performant, and scalable Jax LLM!
Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
An interface library for RL post training with environments.
TPU inference for vLLM, with unified JAX and PyTorch support.
A Python SDK for the Yahoo! Fantasy Sports API
Automate playing Yahoo Fantasy Football
Python API client that can be used to make calls to the Yahoo! Fantasy Sports API. Developed specifically for NHL, but should support the NFL, MLB, and NBA as well.
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
Public repo of projects streamed on twitch.
Inference-time scaling for LLMs-as-a-judge.
A Better UI For Playing the game Connections





