Supporting multiple backends (especially Jax) is nice! Makes experimenting/migra... | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

		dharmeshkakadia on July 11, 2023 \| parent \| context \| favorite \| on: Keras Core: Keras for TensorFlow, Jax, and PyTorch Supporting multiple backends (especially Jax) is nice! Makes experimenting/migrating between them so much more approachable. Any timeline on when can we expect support for distributed Jax training? The doc currently seems to indicate only TF is supported for distributed training.

martin-gorner on July 11, 2023 [–]

Support for distributed JAX training demoed here: bit.ly/keras-on-jax-demo You have to write a custom training loop for now, but it works.

dharmeshkakadia on July 24, 2023 | [–]

Thanks!

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact