Skip to content

GitLab

  • Menu
Projects Groups Snippets
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in
  • N NIFTy
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 17
    • Issues 17
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 12
    • Merge requests 12
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Releases
  • Packages & Registries
    • Packages & Registries
    • Container Registry
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar

On Thursday, 7th July from 1 to 3 pm there will be a maintenance with a short downtime of GitLab.

  • ift
  • NIFTy
  • Merge requests
  • !770

Merged
Created Apr 25, 2022 by David Outland@doutlandDeveloper

sample mapping options

  • Overview 31
  • Commits 16
  • Pipelines 6
  • Changes 3

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.

Assignee
Assign to
Reviewer
Request review from
Time tracking
Source branch: nifty2jax_parallel_mapping