Skip to content

ENH: Add other array frameworks as backends #179

@timmens

Description

@timmens

PyLCM heavily depends on the JAX array framework and many of its amazing operators (e.g., vmap, jit). Additionally it allows for straightforward computation on the GPU.

Other notable array frameworks are pytorch and mlx. Both frameworks provide very similar operators to vmap and jit, and could thus be potentially relatively easy to support or test.

In particular, the mlx framework could benefit rapid development for users on modern MacBooks as it is optimized for Apple Silicon.

Metadata

Metadata

Assignees

No one assigned

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions