maxtext
maxtext copied to clipboard
Use venv and `uv` throughout in lieu of system Python env + `pip`
Description
uv is inconsistently used in the codebase. In addition to being a faster complete rewrite in Rust, uv also has a dependency resolver that always works; whereas the usual pip one is sometimes flakey. And we don't test various Python and uv systems alongside each other anyway, so we might as well just use uv and later we can setup explicit tests to ensure pip works in addition to uv pip.
Tests
CI and manual tests
Checklist
Before submitting this PR, please make sure (put X in square brackets):
- [x] I have performed a self-review of my code. For an optional AI review, add the
gemini-reviewlabel. - [x] I have necessary comments in my code, particularly in hard-to-understand areas.
- [x] I have run end-to-end tests tests and provided workload links above if applicable.
- [x] I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.