Wing Lian cb03c765a1 add uv tooling for e2e gpu tests (#2750) ...

* add uv tooling for e2e gpu tests

* fixes from PR feedback

* simplify check

* fix env var

* make sure to use uv for other install

* use raw_dockerfile_image

* Fix import

* fix args to experimental dockerfile image call

* use updated modal versions

2025-06-05 07:25:06 -07:00

..

__init__.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cicd.sh

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cleanup.py

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

cleanup.sh

Various fixes for CI, save_only_model for RL, prevent packing multiprocessing deadlocks (#2661)

2025-05-12 10:51:18 -04:00

Dockerfile-uv.jinja

add uv tooling for e2e gpu tests (#2750)

2025-06-05 07:25:06 -07:00

Dockerfile.jinja

Sequence parallelism (#2412)

2025-03-21 12:43:55 -04:00

e2e_tests.py

update doc and use P2P=LOC for brittle grpo test (#2649)

2025-05-12 14:17:25 -04:00

multigpu.py

add uv tooling for e2e gpu tests (#2750)

2025-06-05 07:25:06 -07:00

multigpu.sh

update trl to 0.17.0 (#2560)

2025-04-27 19:19:53 -04:00

single_gpu.py

add uv tooling for e2e gpu tests (#2750)

2025-06-05 07:25:06 -07:00