martin-gorner

↗ HN profile [ 44.3 ms ] full profile

Karma: 2
Created: July 12, 2023 (3y ago)
Submissions: 0

demo on JAX: https://bit.ly/keras-on-jax-demo

3y ago · view thread · ↗
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.

3y ago · view thread · ↗