maxtext
maxtext copied to clipboard
Enable llama3.1 405B GPU Training
Here is the minimum config required to support llama3.1 405B GPU Training.