BriefGPT.xyz
Oct, 2021
利用可调用 PyTrees 和筛选变换在 JAX 中实现神经网络 Equinox
Equinox: neural networks in JAX via callable PyTrees and filtered transformations
HTML
PDF
Patrick Kidger, Cristian Garcia
TL;DR
介绍了 Equinox,它是一种小型神经网络库,采用类似 PyTorch 的基于类的方法,同时使用 JAX-like 的函数编程,它通过 PyTree 和函数变换解决了 OO/functional 的差异而不需要引入新的程序抽象。
Abstract
jax
and
pytorch
are two popular Python autodifferentiation frameworks.
jax
is based around pure functions and
→