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
This repository contains three challenges presented on the course of Advanced Algorithms and Parallel Programming (095946) at Politecnico di Milano.
Triton kernels for linear RNN (currently recurrentgemma)
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."