Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
By chatting or signing in you agree to the Terms and chat-message logging (revocable in History).