Simple and fast parallel scan over sequences of tensors, with any binary associative function you specify, for PyTorch (Franz A. Heinsen, 2024)
☆18Apr 15, 2025Updated last year
Alternatives and similar repositories for torch_parallel_scan
Users that are interested in torch_parallel_scan are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Real-Time RTUs☆12Mar 20, 2026Updated 4 months ago
- An official implementation of Looking Beyond Two Frames: End-to-End Multi-Object Tracking Using Spatial and Temporal Transformers☆13Mar 9, 2023Updated 3 years ago
- ☆15May 2, 2026Updated 3 months ago
- Implementation of different Relative Entropy Policy Search flavors☆13Nov 15, 2021Updated 4 years ago
- ☆20Oct 4, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Experiments on the impact of depth in transformers and SSMs.☆46Oct 23, 2025Updated 9 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆63Sep 3, 2025Updated 11 months ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- A package to compute the Continuous Ranked Probability Score (crps) (Matheson and Winkler, 1976; Hersbach, 2000), the fair-crps (fcrps) (…☆22Aug 4, 2022Updated 4 years ago
- Implementation of Continuous Control RL Algorithms☆11Dec 8, 2022Updated 3 years ago
- ☆18Jul 23, 2019Updated 7 years ago
- [KDD 2023] code for "Test accuracy vs. generalization gap: model selection in NLP without accessing training or testing data" https://arx…☆12Oct 17, 2022Updated 3 years ago
- ☆14Mar 4, 2024Updated 2 years ago
- ☆14May 5, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A State-Space Model with Rational Transfer Function Representation.☆84May 17, 2024Updated 2 years ago
- Heuristic Dynamic Programming with Python☆14Jul 28, 2014Updated 12 years ago
- Signax: Signature computation in JAX☆39May 14, 2026Updated 2 months ago
- Standalone implementation of Conformal Prediction and other distribution-free Machine Learning methods.☆15Aug 23, 2019Updated 6 years ago
- Implementation of the fast weight product key memory from Sakana AI☆19Apr 1, 2026Updated 4 months ago
- The Fuzzy-Predictive-Control repository is a comprehensive collection of Python Jupyter Notebooks that demonstrate the implementation of …☆14Aug 7, 2023Updated 3 years ago
- wagwan my slime☆12Nov 24, 2020Updated 5 years ago
- ☆11Sep 9, 2024Updated last year
- A modular landscape evolution modeling package built on top of the Landlab Toolkit.☆21Sep 11, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …☆41Nov 12, 2020Updated 5 years ago
- Exploring how to to deal with uncertain inputs with gaussian process regression models.☆27Apr 16, 2021Updated 5 years ago
- This is the code repository for the DuSt-MPC paper, published at Robotics: Science and Systems (RSS) 2021.☆31Jul 16, 2021Updated 5 years ago
- lanmt ebm☆12Jun 19, 2020Updated 6 years ago
- Source code for the MARL-enabled real-time control of urban drainage system☆18Dec 27, 2024Updated last year
- ☆18Aug 22, 2024Updated last year
- [NeurIPS 2023] Token-Scaled Logit Distillation for Ternary Weight Generative Language Models☆18Dec 6, 2023Updated 2 years ago
- ☆10Jun 28, 2025Updated last year
- ☆23Apr 13, 2024Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Toolbox for working with streaming data as rough paths in Python☆52Jul 1, 2026Updated last month
- Building more accurate protein structures from backbone torsion angles☆16Mar 21, 2025Updated last year
- ☆33Oct 22, 2024Updated last year
- This repository contains a set of scripts for performing singular value decomposition on protein multiple sequence alignments, and analyz…☆11Aug 12, 2024Updated last year
- Official implementation repository for the paper Towards General Conceptual Model Editing via Adversarial Representation Engineering.☆20Dec 6, 2024Updated last year
- 💅 A GitHub Actions workflow template for fine-tuning your own Flux model using the Replicate's API☆30Sep 24, 2024Updated last year
- Distribution and filtering on SO(3) x Euclidean space☆13Jun 24, 2022Updated 4 years ago