https://cluster-site.onrender.com/tags/jax/ Recent content in Jax on Tenu Tech Brief Hugo -- 0.146.0 en-us Wed, 25 Feb 2026 07:59:04 +0000 https://cluster-site.onrender.com/posts/a-developers-guide-to-debugging-jax-on-cloud-tpus-essential-tools-and-techniques/ Tue, 24 Feb 2026 00:39:05 +0000 https://cluster-site.onrender.com/posts/a-developers-guide-to-debugging-jax-on-cloud-tpus-essential-tools-and-techniques/ • JAX on Cloud TPUs provides powerful acceleration for machine learning workflows. • When working in distributed cloud environments, you need specialized tools to debug your workfl https://cluster-site.onrender.com/posts/a-developers-guide-to-debugging-jax-on-cloud-tpus-essential-tools-and-techniques/ Sun, 22 Feb 2026 00:39:25 +0000 https://cluster-site.onrender.com/posts/a-developers-guide-to-debugging-jax-on-cloud-tpus-essential-tools-and-techniques/ • JAX on Cloud TPUs provides powerful acceleration for machine learning workflows. • When working in distributed cloud environments, you need specialized tools to debug your workfl https://cluster-site.onrender.com/posts/accelerating-long-context-model-training-in-jax-and-xla/ Tue, 03 Feb 2026 17:30:00 +0000 https://cluster-site.onrender.com/posts/accelerating-long-context-model-training-in-jax-and-xla/ • Large language models (LLMs) are rapidly expanding their context windows, with recent models supporting sequences of 128K tokens, 256K tokens, and beyond. • However, training the