Skip to content

Speedup jax with vmap #18

@edgarWolf

Description

@edgarWolf

Currently, the optimization problems solved in JaxCurveGenerationSolver are solved sequentially.
One might implement a vectorized version of this using jax.vmap resulting in a speedup and leveraging the whole computation to the XLA-compiled code.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions