ParentFull threadauraham·Where is the GPT implementation in JAX? I only found this [1] in PyTorch and NumPy.[1] https://github.com/karpathy/nanoGPTView on HN