pytorch_fft
by locuslab
PyTorch wrapper for FFTs
AI summary
FFT transformer
Provides an efficient wrapper around CUDA FFTs for PyTorch transformations
- stars
- 315
- forks
- 47
- watching
- 9
Similar projects
Found by comparing what the projects do, not just their names.
Translation model
Implementation of a transformer-based translation model in PyTorch
FFT library
Fast Fourier transform implementation in MicroPython for embedded systems
QANet impl
An implementation of QANet with PyTorch for text classification tasks
Factorization machine library
PyTorch implementation of a Factorization Machine module
Deep Learning Wrapper
Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages.
Transformer model
Implementing OpenAI's transformer language model in PyTorch with pre-trained weights and fine-tuning capabilities
Transformer library
An implementation of transformer models in PyTorch for natural language processing tasks
Transformer model
An implementation of Reformer, an efficient Transformer model for natural language processing tasks.
Optimization layer
A PyTorch module that adds differentiable optimization as a layer to neural networks
Segmentation library
A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models.
Model converter
Converts PyTorch models to their Torch equivalents.
Text Generator
An implementation of the GPT-2 language model in PyTorch for generating text
GA trainer
Trains PyTorch models using a genetic algorithm
C++ extension library
Enables the creation of custom C++ extensions with CUDA support in PyTorch
pytorch/tnt1.7K
PyTorch toolkit
A collection of tools and utilities for building and training neural networks with PyTorch.
