Equinox

github.comtool

Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.

from

JAX

added

2026-10-10

likes

0

JAX › Libraries: “Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.”