Jax - DEV Community

👋 Sign in for the ability to sort posts by relevant, latest, or top.

xbill for Google Developer Experts

Aug 31

Gemma 4 in Pure JAX: What Changes Between Turing and Ada, and What Doesn't

#jax #gemma #cuda #machinelearning

8 min read

xbill for AWS Community Builders

Aug 31

g5g vs g6 for LLM Serving: the Same Code, and 3.7x the Throughput

#aws #jax #cuda #machinelearning

6 min read

xbill for Google Developer Experts

Aug 29

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

#gemma #jax #googleai #machinelearning

6 min read

xbill for Google Developer Experts

Aug 19

Deploying a QAT Checkpoint Your Serving Stack Can't Load: Gemma 4 E2B in Pure JAX on One TPU

#tpu #jax #llm #quantization

10 min read

$refractionray profile$

Shixin Zhang

Jul 15

Large-Scale TensorCircuit Contractions on GPUs: Disabling XLA GPU Autotuning

#jax

4 min read

Goodput vs throughput trade-offs in agent loops

xbill for Google Developer Experts

Jul 29

One TPU Chip, Eight Agents: Serving Small Agent Workloads with Raw JAX

#tpu #llm #jax #agents

15 min read

Madhumitha Kolkar

Aug 4

I Rewrote My RL Agent in DeepMind's Neural Network Library - Day 9 (Haiku)

#jax #reinforcementlearning #machinelearning #deeplearning

3 min read

$refractionray profile$

Shixin Zhang

Jun 12

PyTrees Are Not One Thing: JAX, PyTorch, and TensorFlow Compared

#jax #torch #tensorflow

8 min read

$refractionray profile$

Shixin Zhang

Jun 6

Why JAX Is a Much Better Backend for Quantum Circuit Simulation Than PyTorch

#jax #pytorch

3 min read

blade dancer

Jun 11

You Can Now Train Quantum Signal Processing Phase Angles with Gradient Descent

#quantum #qml #jax #qsp

3 min read

👋 Sign in for the ability to sort posts by relevant, latest, or top.