Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
-
Updated
Jun 4, 2024 - Python
Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
A simple implementation of parallel scan over sequences of tensors, with custom binary associative functions, for PyTorch (Franz A. Heinsen, 2024)
Triton kernels for linear RNN (currently recurrentgemma)
This repository contains three challenges presented on the course of Advanced Algorithms and Parallel Programming (095946) at Politecnico di Milano.
A JAX / NNX implementation of a linear GRU for language modeling
Add a description, image, and links to the parallel-scan topic page so that developers can more easily learn about it.
To associate your repository with the parallel-scan topic, visit your repo's landing page and select "manage topics."