search

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

Using jax test_utils for dtype-aware test tolerances #30

Closed dfm closed 1 year ago

dfm commented 1 year ago

This should provide somewhat more robust floating point comparisons that handle dtypes appropriately.