Skip to content

Move to Flax for autdiff and jit

Eduardo Pignatelli requested to merge flax into main

Created by: epignatelli

Fixes #1 (closed) Fixes #2 (closed) Fixes #3 (closed) Fixes #10 (closed)

Merge request reports