Multi host & sharding support for keras 3.0#21786

Open

amitsrivastava78 wants to merge 9 commits intokeras-team:masterfrom

amitsrivastava78:multi-host-sharding-support

Commits on Oct 24, 2025

Improve OrbaxCheckpoint implementation
amitsrivastava78
committed
Fix code formatting and remove unused variable
amitsrivastava78
committed
Add OrbaxCheckpoint callback with conditional exports and improved test handling
amitsrivastava78
committed

Commits on Oct 27, 2025

Features: Add sharding and multi-host support for JAX backend
amitsrivastava78
committed
Fix OrbaxCheckpoint sharding and multi-host issues
amitsrivastava78
committed