[2111.00254] Equinox: neural networks in JAX via callable PyTrees and filtered transformations - discu.eu

Linking pages

GitHub - patrick-kidger/equinox: Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/ https://github.com/patrick-kidger/equinox 25 comments

Related searches:

Search whole site: site:arxiv.org

Search title: [2111.00254] Equinox: neural networks in JAX via callable PyTrees and filtered transformations

See how to search.

Submit link to: