Starred repositories
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…
Code for paper "Fetuses Made Simple: Modeling and Tracking of Fetal Shape and Pose" (MICCAI 2025)
An open-source AI agent that brings the power of Gemini directly into your terminal.
Lightweight coding agent that runs in your terminal
Pytorch implementation for the paper titled "SimpleAR: Pushing the Frontier of Autoregressive Visual Generation"
Official Repo for Open-Reasoner-Zero
Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.
A Cursor-inspired AI assistant for Obsidian that offers smart autocomplete and interactive chat with your selected notes
Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...
Plugin to improve viewing Markdown files in Neovim
Chat with GitHub Copilot in Neovim
Fully featured & enhanced replacement for copilot.vim complete with API for interacting with Github Copilot
All-in-one LLM CLI tool featuring Shell Assistant, Chat-REPL, RAG, AI Tools & Agents, with access to OpenAI, Claude, Gemini, Ollama, Groq, and more.
Vim plugin for LLM-assisted code/text completion
LOFT: A 1 Million+ Token Long-Context Benchmark
[ICML'24 Oral] "MagicLens: Self-Supervised Image Retrieval with Open-Ended Instructions"
LSP-AI is an open-source language server that serves as a backend for AI-powered functionality, designed to assist and empower software engineers, not replace them.
Video+code lecture on building nanoGPT from scratch
A set of tools to visualize and interact with sequences of 3D data.
Implementation of Nougat Neural Optical Understanding for Academic Documents
Development repository for the Triton language and compiler
A library for differentiable robotics on manifolds.