Picture for Cristian Garcia

Cristian Garcia

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

Add code
Oct 30, 2021
Viaarxiv icon