Zhixuan Lin

forgetting-transformer. [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning

SPACE. Official PyTorch implementation of "SPACE: Unsupervised Object-Oriented Scene Representation via Spatial Attention and Decomposition"

IODINE. PyTorch re-implementation of Multi-Object Representation Learning with Iterative Variational Inference

G-SWM. Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"

cs285-fall-2019. My solutions to CS285 2019 Fall of UC Berkeley

SQAIR-pytorch. PyTorch implementation of Sequential Attend Infer Repeat

gla-jax. Implementation of Gated Linear Attention (GLA) kernels in JAX and pallas.

deep-fp. Website fingerprinting based on CNN

arctic-fox. Python

AIR-pytorch. Pytorch Implementation of Attend, Infer, Repeat

dreamer-pytorch. IFT6163 course project, PyTorch reimplementation of the Dreamer model

homework_fall2020. Assignments for Berkeley CS 285: Deep Reinforcement Learning (Fall 2020)

dotfiles. My dotfiles

LM-RL-finetune. Jupyter Notebook

paper-notes. Paper reading notes

MiniSQL. A Mini Database Management System based on C++

ift-6269-vq-vae. Python

minicad. A mini CAD application in Java

discrete-ssl. Python