Post

@gilesthomas.com

A bit of a debugging journey with JAX, Flax and NNX: www.gilesthomas.com/2026/06/hash...

gilesthomas.comFlax debugging: making a hash of thingsA quick way to help with debugging plumbing issues in JAX training loops.

17/6/26, 2:11 am0 reposts2 likes

No replies yet.