Equinox: Neural networks in Jax via callable PyTrees and filtered transformation | Dark Hacker News