amir-saadat / equinox Goto Github PK
View Code? Open in Web Editor NEWThis project forked from patrick-kidger/equinox
Callable PyTrees and filtered transforms => neural networks in JAX. https://docs.kidger.site/equinox/
License: Apache License 2.0