Add maxtext for arm64 dockerfile.

NVIDIA / JAX-Toolbox

JAX-Toolbox

Apache License 2.0

210 stars 35 forks source link

Closed nouiz closed 2 weeks ago

nouiz commented 3 weeks ago

Only needed special handling for tensorflow-text is needed. Othewise, the 2 files are the same.

gpupuck commented 2 weeks ago

@DwarKapex @olupton Please help review.