Equinox: Neural networks in Jax via callable PyTrees and filtered transformation(arxiv.org)3 points by teddykoker 4 years ago | 0 commentsNo comments yet