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:
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆25Jun 6, 2024Updated 2 years ago
- ☆15May 2, 2026Updated 5 months ago
- Official code repo for paper "Great Memory, Shallow Reasoning: Limits of kNN-LMs"☆24Apr 30, 2025Updated last year
- ☆40Jan 5, 2024Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆48Oct 23, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Transformer for time series forecasting☆11Jun 7, 2022Updated 4 years ago
- solver for discrete Mixed Observable Markov Decision Processes☆11Oct 30, 2020Updated 6 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆65Sep 3, 2025Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆97Dec 5, 2024Updated last year
- A package to compute the Continuous Ranked Probability Score (crps) (Matheson and Winkler, 1976; Hersbach, 2000), the fair-crps (fcrps) (…☆21Aug 4, 2022Updated 4 years ago
- Verify published packages against their source code.☆11Oct 21, 2019Updated 7 years ago
- Implementation of Continuous Control RL Algorithms☆11Dec 8, 2022Updated 3 years ago
- ☆22Aug 17, 2026Updated 2 months 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 4 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Monitoring a PyTorch Lightning CNN with Weights & Biases☆15Jul 26, 2021Updated 5 years ago
- A State-Space Model with Rational Transfer Function Representation.