3 ms·
Where is the GPT implementation in JAX? I only found this [1] in PyTorch and NumPy. [1] https://github.com/karpathy/nanoGPT https://github.com/karpathy/nanoGPT
by auraham 3y ago
Where is the GPT implementation in JAX? I only found this [1] in PyTorch and NumPy.
[1] https://github.com/karpathy/nanoGPT https://github.com/karpathy/nanoGPT