sample mapping options
Compare changes
Beginning with 2nd June, only the "Single Sign On" option for login to the GitLab web interface will be possible. If you don't have an MPCDF wide second factor so far, please get one at our SelfService (https://selfservice.mpcdf.mpg.de). The GitLab internal second factor will not work.
Implementation of sample mapping options for MetricKL() and for mean_value_and_grad(). Pre-implemented options selectable with string keys include jax.lax.map and jax.pmap, the latter for parallel mapping. Other mapping options beyond the pre-implemented ones can be directly passed as the corresponding function itself. Details on how to do this are provided in the documentation.