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.