Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.

Fast, differentiable sorting and ranking in PyTorch

Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)

Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in OpenAI gym

[COLM 2025] Open-Qwen2VL: Compute-Efficient Pre-Training of Fully-Open Multimodal LLMs on Academic Resources

Implementation of the Tolman Eichenbaum Machine in pytorch

[NeurIPS 2022] Zero-Shot Video Question Answering via Frozen Bidirectional Language Models

An unofficial implementation of "Learning-based Video Motion Magnification" in Pytorch.

Official code repository for Prompt-DT.

Code for the paper "On the Adversarial Robustness of Visual Transformers"

A Python package for analyzing and transforming neural latent spaces.

An unofficial implementation for online decision transformer

PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"

Generic template to bootstrap your Python project.

Open-Vocabulary Video Question Answering: A New Benchmark for Evaluating the Generalizability of Video Question Answering Models (ICCV 2023)

Code for the experiments and websites of the paper "Same Task, Different Circuits"

Code for "ResiDual Transformer Alignment with Spectral Decomposition", TMLR 2025