fixed order of parameters in stateless_apply in JAX distributed example#458

Merged

martin-gorner merged 5 commits intomainfrom mgorner-distrib-example

Jul 12, 2023

+2-8

Commits on Jun 21, 2023

added Jax distributed training exammple using a Keras model
martin-gorner
committed

Commits on Jun 22, 2023

Commits on Jul 12, 2023

the order of arguments in stateless_appply has changed. Fixed example.
martin-gorner
committed
Merge branch 'main' into mgorner-distrib-example
martin-gorner
authored