Engineering posts about Jax

Curated summaries and key learnings for engineers working with Jax.

Easy FunctionGemma finetuning with Tunix on Google TPUs

This article discusses the process of fine-tuning the FunctionGemma language model using the Tunix library on Google TPUs. It begins by outlining the capabilities of FunctionGemma as a small language...

Google

A Developer's Guide to Debugging JAX on Cloud TPUs: Essential Tools and Techniques

This article serves as a comprehensive guide for developers working with JAX on Cloud TPUs, focusing on the essential tools and techniques for debugging and profiling machine learning workflows. It...