Skip to content

solve deprecated jax.tree_leaves

Vicentini Filippo requested to merge pv/deprecation into master

Created by: PhilipVinc

Avoids deprecation warning raised by jax

Merge request reports