AINewsnow

Gemma 4 in Pure JAX: What Ports from TPU to GPU, and What Doesn't

This story is from 2026-08-29. It is preserved in the archive; the latest stories are on the live feed.

This article is about running a hand-written Gemma 4 port in pure JAX on three different accelerators, and about the two places the abstraction leaks. The code is here: github.com/xbill9/gemma4-dev What is this project trying to Do? This project aims to serve one Gemma 4 checkpoint from one JAX por…

Read the full story at DEV Community — Machine Learning ↗

Timeline · 1 report

  1. 2026-08-29 00:39 · DEV Community — Machine Learning
    Gemma 4 in Pure JAX: What Ports from TPU to GPU, and What Doesn't

More stories

  1. Google Joins OpenAI, Anthropic, Meta in Disclosing AI Hacks — Bloomberg AI
  2. Anthropic adds support for the AGENTS.md instructions spec to Claude Code; OpenAI contributed AGENTS.md to the Agentic AI Foundation last year (Thomas Claburn/The Register) — Techmeme
  3. Alibaba ships Qwen3.8-Omni-Flash to watch, listen and call tools — r/LocalLLM
  4. Introducing Kimi K3 on Amazon Bedrock — AWS Machine Learning Blog
  5. Introducing Amazon SageMaker HyperPod Inference Gateway — AWS Machine Learning Blog
  6. Introducing Astra for Law — OpenAI News
  7. Newsom signs executive order to explore new AI rules, consider ‘kill switch’ — Politico Technology
  8. Anthropic, OpenAI, SpaceXAI, Google sued over call to ‘pace’ AI development — Politico Technology

Get the daily brief of stories like this at 6:30 every morning →