Equinox: neural networks in JAX via callable PyTrees and filtered transformations

Add code
Oct 30, 2021

Share this with someone who'll enjoy it:

View paper onarxiv icon

Share this with someone who'll enjoy it: