Ax
Ax copied to clipboard
Submitit ask-tell tutorial
Adds a tutorial that shows how to use the AxClient to schedule jobs and tune parameters on a Slurm cluster via SubmitIt.
Code changes:
- The tutorial notebook
- Linking the tutorial notebook for the website. (index.js, tutorials.json)
Testing: Q: How can I make sure that this tutorial shows up correctly on the Ax doc site?
Looks great! A couple notes (that I will just leave here since commenting on a .ipynb is annoying):
- I can handle making sure this renders properly on the website. It's a little easier for me to do this once things have landed in fbsource so let's just get this in as is for now
- Let's use
exp_to_df(ax_client.experiment)
instead ofax_client.generation_strategy.trials_as_df
-- marginally better UX imo - Can we add a cell where we plot the response surface and optimization trace like we do here: https://ax.dev/tutorials/gpei_hartmann_service.html#6.-Plot-the-response-surface-and-optimization-trace ?
- All the tutorials run locally during our testing, so can we comment out
cluster='slurm'
and replace withcluster='local'
and a comment explaining that the cluster is set to local for testing purposes
One last thing: no need to write the covariance since the demo problem only has one metric
Thanks so much for the comments! I fixed the nits and I did another grammar pass.
@mhavasi has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.
@mhavasi merged this pull request in facebook/Ax@894d392dd0a3c42f1f2758c9d5f4923af5a9ebc4.
This is great, thanks for adding this!