George Dahl

### Description I tried searching on https://jax.readthedocs.io/en/latest/search.html?q=checkpoint_dots for checkpoint_dots but couldn't find any documentation. The documentation of jax.checkpoint is helpful, but should provide more information on the predefined policies in...

It would be really useful to have the pinball loss defined. Similar to sklearn https://scikit-learn.org/stable/modules/generated/sklearn.metrics.mean_pinball_loss.html

Does this code come with a License? If so, could you add a license file? If you have no strong opinion and do intend to provide it under an open...

In submission_runner.py, if we are in the self-tuning rules, the hyperparameters argument to train_once will always be None. Then in this code snippet ``` dropout_rate = None aux_dropout_rate = None...