Callable PyTrees and filtered transforms => neural networks in JAX. https://docs.kidger.site/equinox/
Primary LanguagePythonApache License 2.0Apache-2.0
No one’s star this repository yet.