Provide PyTorch implementations by wrapping JAX functions#277

Merged

matt-graham merged 52 commits intomainfrom

mmg/pytorch-wrapper

Apr 23, 2025

Commits on Feb 24, 2025

Make spherical precompute benchmarks compatible with pytorch
matt-graham
committed

Commits on Feb 26, 2025

Add utilities for wrapping JAX function for use in PyTorch
matt-graham
committed
Add initial wrapped versions of torch precompute transforms
matt-graham
committed

Commits on Mar 6, 2025

Commits on Mar 11, 2025

Commits on Mar 20, 2025

Commits on Mar 28, 2025

Commits on Apr 3, 2025

Merge branch 'main' into mmg/pytorch-wrapper
matt-graham
committed

Commits on Apr 4, 2025

Commits on Apr 8, 2025

Add test for function checking torch available
matt-graham
committed

Commits on Apr 11, 2025

Fix torch optional import logic to avoid errors when not installed
matt-graham
committed

Commits on Apr 16, 2025

Commits on Apr 23, 2025