Browsing Tag
jax
9 posts
Fine-Tuning Gemma 3 on TPU for Medical Q&A with Keras and JAX
I wanted to build a medical Q&A fine-tuning project that stayed genuinely TPU-native on Kaggle. This project uses…
Building a Nano Mixture-of-Experts (MoE) Language Model in JAX from Scratch
A beginner-friendly deep dive into how Mixture-of-Experts works, why it matters, and how to build one in pure…
Optimizing Data Loading Performance in JAX with jax-dataloader and Grain
1. Executive Summary JAX, a high-performance numerical computation library, is celebrated for its XLA-accelerated compute capabilities, which often…
JAX DataLoader v0.1.9: A High-Performance Data Loading Solution for JAX
🔥 Introduction Loading data efficiently is one of the biggest challenges in machine learning. If you’ve been working…
Building an Efficient DataLoader in JAX: From Scratch to Production
Introduction JAX is a powerful framework for numerical computing, enabling high-performance machine learning and deep learning. However, one…
Machine Learning Communities: Q2 ‘23 highlights and achievements
Posted by Nari Yoon, Bitnoori Keum, Hee Jung, DevRel Community Manager / Soonson Kwon, DevRel Program Manager Let’s…
Machine Learning Communities: Q1 ‘23 highlights and achievements
Posted by Nari Yoon, Bitnoori Keum, Hee Jung, DevRel Community Manager / Soonson Kwon, DevRel Program Manager Let’s…
Machine Learning Communities: Q4 ‘22 highlights and achievements
Posted by Nari Yoon, Hee Jung, DevRel Community Manager / Soonson Kwon, DevRel Program Manager Let’s explore highlights…
Machine Learning Communities: Q3 ‘22 highlights and achievements
Posted by Nari Yoon, Hee Jung, DevRel Community Manager / Soonson Kwon, DevRel Program Manager Let’s explore highlights…