tucan9389/nanochat-jax Discussion: Porting nanochat to a TPU: what carries over from PyTorch, and what breaks #1

github.com · 2026-07-15 · by Long Nguyen ·github

Porting nanochat to a TPU: what carries over from PyTorch, and what breaks

tucan9389/nanochat-jax

Karpathy's nanochat normally runs on an 8×H100 GPU node, and several ports of it to JAX already exist. Among them, my aim was to keep the config and architecture as close to nanochat as possible (p...

source

github.com · 7 images · 162 links

Comments (0)

No comments yet