Skip to content

Commit

Permalink
Jax tracing fix (#20412)
Browse files Browse the repository at this point in the history 
* `JAXTrainer`: refactoring and fixes
Fix for #20402
Fix for #20411

* CI setup

* Fix tests

* Revert CI branch to master

* `function` -> `iterator_step`
  • Loading branch information
@nicolaspi
nicolaspi authored Oct 26, 2024
1 parent 093b055 commit fcb3c27
Show file tree
Hide file tree
Showing 4 changed files with 372 additions and 184 deletions.
Loading

0 comments on commit fcb3c27

Please sign in to comment.