research
∙
10/30/2021
Equinox: neural networks in JAX via callable PyTrees and filtered transformations
JAX and PyTorch are two popular Python autodifferentiation frameworks. J...
research
∙
02/05/2020