Skip to content

Replace jax.tree_util.tree_map() with jax.tree_util.tree_multimap()#3

Open
oikosohn wants to merge 1 commit intogordicaleksa:mainfrom oikosohn:oikosohn-patch-1
+6-6

Commits

Commits on Nov 29, 2023