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 NUFFT implementation is that of the Flatiron Institute. Please refer to the FINUFFT and cuFINUFFT repositories for details.
Installation
You can install TensorFlow NUFFT with pip:
$ pip install tensorflow-nufft
Note that only Linux wheels are currently being provided.
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 our Zenodo record.
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.3.0b0-cp39-cp39-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | df71694a687b3196bb1f032ae83ca32d4309f096b55bacbde2a12daeeb8e04c3 |
|
MD5 | 07be00f194f98d78ea13b172c39c61c8 |
|
BLAKE2b-256 | f27e51c0f04e85c5a6f29f7cf23219f2987333c9ca21e92d4eaf2d2434f0c1b0 |
Hashes for tensorflow_nufft-0.3.0b0-cp38-cp38-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | f250708a16081990e1a1901a617bb372cc21efb308e77b4dee1dd68ab996e95c |
|
MD5 | d88f84356cce172334772b4c966c6936 |
|
BLAKE2b-256 | 1646d5a8e3a0e835735cb2c22c0bdbc4e4563ae854036fcbb6435ed80fb39a4f |
Hashes for tensorflow_nufft-0.3.0b0-cp37-cp37m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 12b6d6b6786132c3d09edb2d3108a92c0e35eec3afd124451b6f0fa981e51bb0 |
|
MD5 | 0405ee9b0b423c30123459c4618c4afe |
|
BLAKE2b-256 | 13bc31f1ad3f383301d887912e86b0cd87dad20bc0a130e15eee697cf7eba994 |
Hashes for tensorflow_nufft-0.3.0b0-cp36-cp36m-manylinux_2_12_x86_64.manylinux2010_x86_64.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0d0b8b091dce1079b3f08f63fb63ffe5c0dc161fac93181734f2e28a9d3cf697 |
|
MD5 | 8303fe26c91b49e2c1d036aa66ef186d |
|
BLAKE2b-256 | 6bb4da6719715b794856c5bd42512a362dc9a08a79673a0417b16ec4ecdb3793 |