4 ms·Equinox: Neural networks in Jax via callable PyTrees and filtered transformation3 points by teddykoker 4y ago