A fast, native non-uniform FFT op for TensorFlow.
Project description
TensorFlow NUFFT is a fast, native non-uniform fast Fourier transform op for TensorFlow. It provides:
Fast CPU/GPU kernels. The TensorFlow framework automatically handles device placement as usual.
A simple, well-documented Python interface.
Gradient definitions for automatic differentiation.
Shape functions to support static shape inference.
The underlying algorithm is based on the NUFFT implementation by the Flatiron Institute. Please refer to FINUFFT and cuFINUFFT for more details.
Installation
You can install TensorFlow NUFFT with pip:
$ pip install tensorflow-nufft
Note that only Linux wheels are currently being provided.
TensorFlow Compatibility
Each TensorFlow NUFFT release is compiled against a specific version of TensorFlow. Please see the compatibility table below to see what versions of each package you can expect to work together.
TensorFlow MRI Version |
TensorFlow Compatibility |
Release Date |
---|---|---|
v0.5.0 |
v2.7.x |
Nov 12, 2021 |
v0.4.0 |
v2.7.x |
Nov 8, 2021 |
v0.3.2 |
v2.6.x |
Aug 18, 2021 |
v0.3.1 |
v2.6.x |
Aug 18, 2021 |
v0.3.0 |
v2.6.x |
Aug 13, 2021 |
Documentation
Visit the docs for the API reference and examples of usage.
Contributions
If you use this package and something does not work as you expected, please file an issue describing your problem. We will do our best to help.
Contributions are very welcome. Please create a pull request if you would like to make a contribution.
Citation
If you find this software useful in your work, please cite us.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Hashes for tensorflow_nufft-0.5.0-cp39-cp39-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dee4dea24e82d983a5691e2e0e26585a151a4a0184e89ed7f083b336b4735ce8 |
|
MD5 | 754276af36d306a3d87117dab5b84e6a |
|
BLAKE2b-256 | 821d12411633227331b035592ff2e3f6e7ba3fe2ff5b014b9954075dc15a7f82 |
Hashes for tensorflow_nufft-0.5.0-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fcf6f29f9cd2e3db716f9dfe8f00250cccd26f5ad7e572b4b4a819848d9ce665 |
|
MD5 | c650a6711ac520dd1c191cf970bb17ca |
|
BLAKE2b-256 | 9b93478a6f1f0bc6577881801cabf90e7cab52e386795f75157f06159d86287d |
Hashes for tensorflow_nufft-0.5.0-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0cf620844e6712d5e9c91d9af6fccdfcdc4edd34320e05776a566fe185b8d631 |
|
MD5 | dca4e3c1ebcdfc55a16c8ae236ace180 |
|
BLAKE2b-256 | 94971784363ff9df82e9a6cc8da77a76b2457f13a040ce50b19a5737b74d8e72 |