Use the JTE API to set stacked table stats...#169

Merged

hertschuh merged 2 commits intokeras-team:mainfrom

Oct 30, 2025

Commits on Oct 30, 2025

Use the JTE API to set stacked table stats to the maximum of the input table specs.
cantonios
committed
Update keras_rs/src/layers/embedding/jax/distributed_embedding.py

cantonios
and
gemini-code-assist[bot]
authored