Code

Jax tracing fix#20412

Code

Merged

fchollet merged 6 commits intokeras-team:masterkeras-team/keras:masterfrom nicolaspi:jax-tracing-fixnicolaspi/keras:jax-tracing-fixCopy head branch name to clipboardOct 26, 2024

+372-184

Commits

Commits on Oct 25, 2024

• JAXTrainer: refactoring and fixes

  Show description for a49d443
  nicolaspi

  committed
  a49d443

  Copy full SHA for a49d443

• CI setup

  nicolaspi

  committed
  09e85f1

  Copy full SHA for 09e85f1

• Fix tests

  nicolaspi

  committed
  cec88d5

  Copy full SHA for cec88d5

• Revert CI branch to master

  nicolaspi

  committed
  523b093

  Copy full SHA for 523b093

Commits on Oct 26, 2024

• Merge branch 'keras-team:master' into jax-tracing-fix

  nicolaspi

  authored
  e30714f

  Copy full SHA for e30714f

• function -> iterator_step

  nicolaspi

  committed
  fe397ed

  Copy full SHA for fe397ed