Dev
GitHub repos gaining traction - what high-signal users are starring and what's climbing the board, captured daily and enriched from GitHub. Raw material for spotting new tech and patterns worth building on.
2,235
repos tracked
175
surfaced this week
194
created < 30d
Python
top language
23 repos
-
JAX/Equinox implementation of ProteinMPNN and LigandMPNN for protein sequence design
-
JAX in JavaScript – ML library for the web, running on WebGPU & Wasm
-
Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.
-
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
A multi-framework neural network compiler lowering PyTorch, JAX, Flax, and Keras models to portable, high-performance GGML execution
-
A Pure JAX AsyncRL stack
-
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
-
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
-
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
-
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
-
A retargetable MLIR-based machine learning compiler and runtime toolkit.
-
marimo + uv + jax starter template
-
CUDA Craftax-Classic: 7x faster RL training than JAX
-
A Python package for probabilistic state space modeling with JAX
-
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
-
A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch
-
Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.
-
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
-
Student materials for DIS 2026 JAX-on-TPU workshop
-
Code release for "i1: A Simple and Fully Open Recipe for Strong Text-to-Image Models"