Fast Discounted Cumulative Sums in PyTorch
☆96Aug 28, 2021Updated 4 years ago
Alternatives and similar repositories for torch-discounted-cumsum
Users that are interested in torch-discounted-cumsum are comparing it to the libraries listed below
Sorting:
- To be a next-generation DL-based phenotype prediction from genome mutations.☆18May 17, 2021Updated 4 years ago
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆55Sep 18, 2022Updated 3 years ago
- ☆80Jan 21, 2022Updated 4 years ago
- Replication attempt for the Protein Folding Model described in https://www.biorxiv.org/content/10.1101/2021.08.02.454840v1☆36May 19, 2022Updated 3 years ago
- Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch☆12Jan 16, 2022Updated 4 years ago
- CLASP - Contrastive Language-Aminoacid Sequence Pretraining☆142Sep 17, 2021Updated 4 years ago
- Python Research Framework☆106Nov 3, 2022Updated 3 years ago
- ☆21Mar 15, 2023Updated 2 years ago
- ☆26May 9, 2022Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- ☆29Oct 3, 2022Updated 3 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆132Aug 6, 2022Updated 3 years ago
- Contrastive Language-Image Pretraining☆144Sep 6, 2022Updated 3 years ago
- Code for "Bridging the Gap between f-GANs and Wasserstein GANs", ICML 2020☆14Jul 18, 2020Updated 5 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Dec 21, 2021Updated 4 years ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆37Jan 8, 2026Updated 2 months ago
- Hidden Engrams: Long Term Memory for Transformer Model Inference☆35Jun 26, 2021Updated 4 years ago
- Model-Based Reinforcement Learning via Latent-Space Collocation.☆34Mar 29, 2023Updated 2 years ago
- Contrastive Language-Audio Pretraining☆87Mar 6, 2022Updated 4 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 10 months ago
- pointMass pybullet RL environment for simple experiments☆23Jan 12, 2022Updated 4 years ago
- Pedagogical codebase for a simplified score-based generative model design, with training loop☆40Aug 28, 2021Updated 4 years ago
- ☆20Mar 14, 2021Updated 4 years ago
- ☆40May 2, 2021Updated 4 years ago
- Code for the paper PermuteFormer☆41Oct 10, 2021Updated 4 years ago
- A Pytree Module system for Deep Learning in JAX☆213Feb 26, 2023Updated 3 years ago
- Decentralized Reinforcment Learning: Global Decision-Making via Local Economic Transactions (ICML 2020)☆43Dec 8, 2022Updated 3 years ago
- The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".☆34Jun 11, 2025Updated 8 months ago
- pytorch implementation of trDesign☆44Mar 19, 2021Updated 4 years ago
- PyTorch optimizers with sparse momentum and weight decay☆10Oct 3, 2020Updated 5 years ago
- ☆11Sep 13, 2022Updated 3 years ago
- Experiments in protein folding through language modeling☆10Dec 10, 2021Updated 4 years ago
- Pytorch-Lightning Seq2seq model with the use of recurrent neural network☆10Mar 29, 2021Updated 4 years ago
- A2C is a special case of PPO!☆22May 20, 2022Updated 3 years ago
- Official repository with code and data accompanying the NAACL 2021 paper "Hurdles to Progress in Long-form Question Answering" (https://a…☆45Jul 30, 2022Updated 3 years ago
- Accelerated First Order Parallel Associative Scan☆195Jan 7, 2026Updated 2 months ago
- Fast Block Sparse Matrices for Pytorch☆549Jan 21, 2021Updated 5 years ago
- A framework for implementing equivariant DL☆10May 25, 2021Updated 4 years ago
- Official repository for the paper "Exploring the Promise and Limits of Real-Time Recurrent Learning" (ICLR 2024)☆13Jun 11, 2025Updated 8 months ago