Tpu training container - Githubissues

huggingface / Google-Cloud-Containers

Hugging Face Deep Learning Containers (DLCs) for Google Cloud

https://hf.co/docs/google-cloud

Apache License 2.0

127 stars 16 forks source link

Tpu training container #38

Closed tengomucho closed 5 months ago

tengomucho commented 5 months ago

Added updated training container for TPU, leveraging work done onoptimum-tpu project. README updated, added the notebook in the container, to provide an example for gemma tuning, tested successfully on the container.