jax
c15d1075 - Merge pull request #32527 from jax-ml:shmap

Commit

120 days ago

Merge pull request #32527 from jax-ml:shmap PiperOrigin-RevId: 817777591

References

#32527 - Use factory decorator pattern in shard_map tutorial

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading