flatironinstitute / jax-finufft

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
Apache License 2.0
80 stars 3 forks source link

Adding more sensible shape handling #15

Closed dfm closed 2 years ago

dfm commented 2 years ago

This should also close #13 and #14, when it's done!