Luke Baumann

Results 8 comments of Luke Baumann

I see this is not the only place this type of check occurs.

> Thank you for the change Luke! I added you to the repo as a writer so you can create a new branch based PR. I think the forking method...

Pathways TPU and McJAX TPU have different dependencies. Specifically `jax[tpu]` is only needed for McJAX TPU and `pathwaysutils` is only needed for Pathways TPU. Separating Pathways TPU and McJAX TPU...

I removed the pathways-tpu image from this PR. I think it is helpful to separate dependencies for McJAX/Pathways in the long term but agree that it is not worth any...

k8s does not respect host ports for init containers when scheduling so we have witnessed collisions where job 1 containers connect to job 2 if the head pods were on...

2 cents: The main issue covered in the link is installing an unmanaged version of gcloud which may only be a problem when running XPK from a GCE instance.

The existing code has a few assumptions about where xpk is installed. In some places, it expects it in `./~/xpk`, in some, `./../xpk`, in some `./xpk` (in others, maybe even...

There are references to an unexpanded tilde in benchmark runner files specifically [benchmark_runner.py](https://github.com/AI-Hypercomputer/maxtext/blob/a386752d28e8ec0b662028a33e91075420e399bc/benchmarks/benchmark_runner.py#L155) [pw_long_running_recipe.py](https://github.com/AI-Hypercomputer/maxtext/blob/a386752d28e8ec0b662028a33e91075420e399bc/benchmarks/recipes/pw_long_running_recipe.py#L43) [maxtext_xpk_runner.py](https://github.com/AI-Hypercomputer/maxtext/blob/a386752d28e8ec0b662028a33e91075420e399bc/benchmarks/maxtext_xpk_runner.py#L102) [pw_mcjax_benchmark_recipe.py](https://github.com/AI-Hypercomputer/maxtext/blob/a386752d28e8ec0b662028a33e91075420e399bc/benchmarks/recipes/pw_mcjax_benchmark_recipe.py#L46) [pw_mcjax_checkpoint_benchmark_recipe.py](https://github.com/AI-Hypercomputer/maxtext/blob/a386752d28e8ec0b662028a33e91075420e399bc/benchmarks/recipes/pw_mcjax_checkpoint_benchmark_recipe.py#L50) @SamuelMarks were there other references that you are aware of?