Refactoring Maxtext build process with stable stack

AI-Hypercomputer / maxtext

A simple, performant and scalable Jax LLM!

Apache License 2.0

1.47k stars 275 forks source link

Closed parambole closed 1 day ago

parambole commented 6 days ago

Added Mode=stable_stack docker_build_dependency_image.sh which enables Maxtext to be built on top of jax stable stack to maintain consistency
Removed the old flow