search

NVIDIA / JAX-Toolbox

JAX-Toolbox
Apache License 2.0
234 stars 44 forks source link

Add maxtext for arm64 dockerfile. #914

Closed nouiz closed 3 months ago

nouiz commented 3 months ago

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

olupton commented 3 months ago

The source branch needs to be in the main repository, not a fork, for the pipeline to work correctly.

nouiz commented 3 months ago

Thanks. Replaced by : https://github.com/NVIDIA/JAX-Toolbox/pull/921