Equinox: Neural networks in Jax via callable PyTrees and filtered transformation #1 Post by teddykoker » Fri, Apr 15, 2022, 5:43 PM UTC Equinox: Neural networks in Jax via callable PyTrees and filtered transformationarxiv.org