EngineerKhan/equinox
Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
PythonApache-2.0
No issues in this repository yet.
Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
PythonApache-2.0
No issues in this repository yet.