Skip to content
View mjsML's full-sized avatar
🔥
SOL
🔥
SOL

Highlights

  • Pro

Organizations

@NVIDIA @probml @jax-ml @openxla @NVIDIA-NeMo

Block or report mjsML

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Content in all repositories owned by your account will be closed.
Maximum 250 characters. Please don’t include any personal information such as legal names or email addresses. Markdown is supported. This note will only be visible to you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
Showing results

ZMK Firmware Repository

C 4,277 4,164 Updated Sep 24, 2026

A vLLM patch + hand‑written SM120 SASS kernels: 2‑bit MoE experts + an FP4 "delta" cache that recovers precision — matching the official (NV)FP4 checkpoint's quality on consumer Blackwell cards

Sass 541 49 Updated Sep 23, 2026

Use Garry Tan's exact Claude Code setup: 23 opinionated tools that serve as CEO, Designer, Eng Manager, Release Manager, Doc Engineer, and QA

TypeScript 134,123 19,983 Updated Sep 25, 2026

💫 Toolkit to help you get started with Spec-Driven Development

Python 138,793 12,426 Updated Sep 24, 2026

Environments and Algorithms for Generative Flow Networks in JAX

Python 92 2 Updated May 10, 2026

RLinf: Reinforcement Learning Infrastructure for Embodied and Agentic AI

Python 5,366 749 Updated Sep 24, 2026

Machine Learning Engineering Open Book

Python 19,039 1,249 Updated Sep 23, 2026

A unofficial desktop web interface for managing Speediance Gym Monster workouts, viewing the exercise library, and generating AI-powered plans.

HTML 46 18 Updated Mar 6, 2026

Miles is an enterprise-facing reinforcement learning framework for LLM and VLM post-training, forked from and co-evolving with slime.

Python 2,983 517 Updated Sep 25, 2026

JAXMg: A multi-GPU linear solver in JAX

C++ 71 4 Updated Sep 24, 2026

Puffing up reinforcement learning

C 6,474 578 Updated Sep 13, 2026
Python 356 73 Updated Sep 24, 2026

Machine Learning Systems: Foundations, Scaling, Agentic AI, and Physical AI (Vols I–IV) • Harvard CS249r | https://mlsysbook.ai

Python 28,433 3,598 Updated Sep 25, 2026

JAX-Toolbox

Python 433 82 Updated Sep 24, 2026

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Python 331 34 Updated Sep 18, 2026

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

Jupyter Notebook 3,192 681 Updated Jun 1, 2026

Minimalistic 4D-parallelism distributed training framework for education purpose

Python 2,311 208 Updated Aug 26, 2025

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

Python 7,145 628 Updated Sep 24, 2026

Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach

Jupyter Notebook 165 32 Updated Mar 15, 2024

Solution Manual for Roman Vershynin's High-Dimensional Probability

TeX 26 2 Updated Jul 26, 2024

A JAX-based simulator for autonomous driving research.

Python 1,103 133 Updated Oct 23, 2025

Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.

Python 1,675 112 Updated Sep 15, 2026

MLIR-based partitioning system

MLIR 213 42 Updated Sep 24, 2026

CloudAI Benchmark Framework

Python 99 62 Updated Sep 23, 2026

seqax = sequence modeling + JAX

Python 199 21 Updated Jul 23, 2025

PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.

Python 1,474 122 Updated Sep 24, 2026

Grok open release

Python 52,233 8,544 Updated Aug 30, 2024

Checkpoint/Restore tool

C 4,003 791 Updated Sep 24, 2026

High-performance key-value store for ML inference. 100x faster than Redis.

Rust 227 3 Updated May 14, 2024

Gemma open-weight LLM library, from Google DeepMind

Python 5,746 1,027 Updated Sep 16, 2026
Next