AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!
Apache License 2.0
1.47k stars 275 forks source link

added run_name_prefix to tensorboard #856

Closed kyle-google closed 2 weeks ago

kyle-google commented 3 weeks ago

DO NOT MERGE THIS, simply an example of maxtext and vertex ai tensorboard integration.