Commit 8c62637
committed
Rework conversion between autograd/jax arrays to avoid xla error in jax>=0.4.3
1 parent f19a3c1 commit 8c62637
File tree
16 files changed
+568
-436
lines changed16 files changed
+568
-436
lines changedSome generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
0 commit comments