Tweeted By @dennybritz
From PyTorch to JAX: towards neural net frameworks that purify stateful code: https://t.co/Sg3k4XpzTD
— Denny Britz (@dennybritz) March 10, 2020
Great writeup from @sjmielke on how to think about JAX programs and how it all works - from scratch.