pytorch_fft

FFT transformer

Provides an efficient wrapper around CUDA FFTs for PyTorch transformations

PyTorch wrapper for FFTs

GitHub

315 stars
9 watching
47 forks
Language: Python
last commit: almost 8 years ago

Related projects:

RepositoryDescriptionStars
leviswind/pytorch-transformerImplementation of a transformer-based translation model in PyTorch240
peterhinch/micropython-fourierFast Fourier transform implementation in MicroPython for embedded systems79
hengruo/qanet-pytorchAn implementation of QANet with PyTorch for text classification tasks345
jmhessel/fmpytorchPyTorch implementation of a Factorization Machine module173
hughperkins/pytorchProvides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages.432
huggingface/pytorch-openai-transformer-lmImplementing OpenAI's transformer language model in PyTorch with pre-trained weights and fine-tuning capabilities1,511
tongjilibo/bert4torchAn implementation of transformer models in PyTorch for natural language processing tasks1,257
lucidrains/reformer-pytorchAn implementation of Reformer, an efficient Transformer model for natural language processing tasks.2,132
locuslab/optnetA PyTorch module that adds differentiable optimization as a layer to neural networks517
yunlongdong/fcn-pytorchA PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models.161
clcarwin/convert_torch_to_pytorchConverts PyTorch models to their Torch equivalents.541
graykode/gpt-2-pytorchAn implementation of the GPT-2 language model in PyTorch for generating text976
ahmedfgad/torchgaTrains PyTorch models using a genetic algorithm96
pytorch/extension-cppEnables the creation of custom C++ extensions with CUDA support in PyTorch1,031
pytorch/tntA collection of tools and utilities for building and training neural networks with PyTorch.1,669