Equinox: Elegant easy-to-use neural networks in Jaxgithub.com/patrick-kidger2 pointskristianp3 years ago