keras-team / keras-core

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
Apache License 2.0
1.27k stars 115 forks source link

Create distribution_lib for TF backend #889

Closed hubingallin closed 11 months ago

hubingallin commented 11 months ago

Add Mesh and Layout helper functions