NanoChat JAX & Flax NNX Port (TPU Sprint)

NanoChat JAX & Flax NNX Port (TPU Sprint) | Omotayo AinaPort of Andrej Karpathy’s NanoChat LLM architecture to JAX and Flax NNX for Google Cloud TPUs.This site needs JavaScript for navigation. Contact: LinkedIn.