I
Jul 20, 2026I Started Learning JAX as a Senior ML Engineer - Here's My First Impression.
SERIES: Learning RL and JAX in Public - from zero to DeepMind ! I have been a machine...
Jul 20, 20263 min read2 reactions0 comments
Back to articles 
Jun 11, 2026
Oct 19, 2024
Tag archive
#jax
L
Jul 15, 2026Large-Scale TensorCircuit Contractions on GPUs: Disabling XLA GPU Autotuning
When running large-scale tensor-network contractions with TensorCircuit-NG and the JAX GPU backend,...
Jul 15, 20264 min read0 reactions0 comments
T
Jul 18, 2026Training a 1,000-Qubit, 40,000-Parameter Quantum Algorithm with Full Gradients Using TensorCircuit-NG
Introduction If you wanted to simulate 1,000 quantum qubits and compute exact full...
Jul 18, 202611 min read0 reactions0 comments
P
Jun 12, 2026PyTrees Are Not One Thing: JAX, PyTorch, and TensorFlow Compared
PyTrees look deceptively simple. You flatten a nested Python object into leaves, keep a structure...
Jun 12, 20268 min read0 reactions0 comments
W
Jun 6, 2026Why JAX Is a Much Better Backend for Quantum Circuit Simulation Than PyTorch
Modern quantum circuit simulation is not just “machine learning with complex tensors.” It involves...
Jun 6, 20263 min read0 reactions0 comments
You Can Now Train Quantum Signal Processing Phase Angles with Gradient Descent
A reproducible benchmark and JAX‑traceable implementation is now available – no analytic solver...
Jun 11, 20263 min read0 reactions1 comments
P
Apr 20, 2026Post-Training Gemma 3 for Earth Observation (EO) Understanding: A JAX Stack + TPU Pipeline for…
Post-Training Gemma 3 for Earth Observation (EO) Understanding: A JAX Stack + TPU Pipeline...
Apr 20, 20267 min read0 reactions0 comments
Bridging numerical relativity and automatic differentiation using JAX
Introduction Albert Einstein's general theory of relativity is the most successful picture...
Oct 19, 202422 min read1 reactions0 comments