3 Pull requests merged by 1 person

• [jax_tpu_embedding] avoid private import of xla.apply_primitive

  #266 merged May 28, 2025

• Change all us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build container to us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build.

  #265 merged May 28, 2025

• Record and use required COO buffer size.

  #243 merged May 27, 2025

5 Pull requests opened by 1 person

• add a use shardy true variant for `trainer_main_sparsecore_test`

  #262 opened May 27, 2025

• Do not count the sentinel node for checking the buffer overflow as it is not actually filled in the COO buffer.

  #263 opened May 27, 2025

• Add the Adam optimizer from [Kingma et al., 2014](http://arxiv.org/abs/1412.6980).

  #264 opened May 28, 2025

• Replace numpy arrays with Eigen Matrix

  #267 opened May 29, 2025

• Add the FTRL (Follow The Regularized Leader) optimizer.

  #268 opened May 29, 2025