Posts tagged JAX A Guide to Implementing and Training Generative Pre-trained Transformers (GPT) in JAX on AMD GPUs 02 July 2024 Douglas Jia English Applications & models AI/ML GenAI JAX PyTorch LLM 2 July, 2024 by Douglas Jia. Read more ... LLM distributed supervised fine-tuning with JAX 25 January 2024 Douglas Jia English Applications & models LLM AI/ML GenAI Fine-Tuning JAX 25 Jan, 2024 by Douglas Jia. Read more ...