pytorch_fft

FFT transformer

Provides an efficient wrapper around CUDA FFTs for PyTorch transformations

PyTorch wrapper for FFTs

GitHub

314 stars
9 watching
47 forks
Language: Python
last commit: about 6 years ago

Related projects:

Repository Description Stars
leviswind/pytorch-transformer Implementation of a transformer-based translation model in PyTorch 239
peterhinch/micropython-fourier Fast Fourier transform implementation in MicroPython for embedded systems 80
hengruo/qanet-pytorch An implementation of QANet with PyTorch for text classification tasks 345
jmhessel/fmpytorch PyTorch implementation of a Factorization Machine module 173
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
huggingface/pytorch-openai-transformer-lm Implementing OpenAI's transformer language model in PyTorch with pre-trained weights and fine-tuning capabilities 1,511
tongjilibo/bert4torch An implementation of transformer models in PyTorch for natural language processing tasks 1,241
lucidrains/reformer-pytorch An implementation of Reformer, an efficient Transformer model for natural language processing tasks. 2,120
locuslab/optnet A PyTorch module that adds differentiable optimization as a layer to neural networks 513
yunlongdong/fcn-pytorch A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models. 160
clcarwin/convert_torch_to_pytorch Converts PyTorch models to their Torch equivalents. 539
graykode/gpt-2-pytorch An implementation of the GPT-2 language model in PyTorch for generating text 973
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm with PyGAD 96
pytorch/extension-cpp Enables the creation of custom C++ extensions with CUDA support in PyTorch 1,017
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,666