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.
Implementation of several algorithms in OpenCL
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."