Explore projects

MatDEM documents

CUDA Templates for Linear Algebra Subroutines

A framework for few-shot evaluation of language models.

Fast Hadamard transform in CUDA, with a PyTorch interface

The official repository of Quamba

Mamba SSM architecture

TensorFlow implementation of "A Relational Intervention Approach for Unsupervised Dynamics Generalization in Model-Based Reinforcement Learning" (ICLR 2022).