Stars
A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.
Post-training with Tinker
🦞 Just talk to your agent — it learns and EVOLVES 🧬.
python driver for tenstorrent blackhole cards
tviz is a visualization dashboard for the Tinker API
Harbor is a framework for running agent evaluations and creating and using RL environments.
Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT device.
torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytor…
torchcomms: a modern PyTorch communications API
Sandbox infrastructure for AI Agents
ray & RLlib tools for unified code across different repositories. Experiments with dynamic hyperparameters
slime is an LLM post-training framework for RL Scaling.
The 100 line AI agent that solves GitHub issues or helps you in your command line. Radically simple, no huge configs, no giant monorepo—but scores >74% on SWE-bench verified!
🤘 TT-NN operator library, and TT-Metalium low level kernel programming model.
An Efficient and User-Friendly Scaling Library for Reinforcement Learning with Large Language Models
A tool for creating and running Linux containers using lightweight virtual machines on a Mac. It is written in Swift, and optimized for Apple silicon.
Scalable toolkit for efficient model reinforcement
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
SkyRL: A Modular Full-stack RL Library for LLMs
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
A fast Common Lisp binary serializer/deserializer
Large Language Model (LLM) Systems Paper List