locuslab

Sequence modeling benchmarks and temporal convolutional networks

Implementation of ConvMixer for "Patches Are All You Need? 🤷"

A fast and differentiable model predictive control (MPC) solver for PyTorch.

A simple and effective LLM pruning approach.

[NeurIPS'19] Deep Equilibrium Models

A fast and differentiable QP solver for PyTorch.

[NeurIPS 2025] The one-stop repository for LLM unlearning

OptNet: Differentiable Optimization as a Layer in Neural Networks

[ICLR'19] Trellis Networks for Sequence Modeling

[ICLR 2020] A repository for extremely fast adversarial training using FGSM

Bridging deep learning and logical reasoning using a differentiable satisfiability solver.

Provable adversarial robustness at ImageNet scale

A method for training neural networks that are provably robust to adversarial attacks.

Consistency Models Made Easy

Input Convex Neural Networks

PyTorch wrapper for FFTs

A differentiable LCP physics engine in PyTorch.

[NeurIPS'20] Multiscale Deep Equilibrium Models

Task-based end-to-end model learning in stochastic optimization

Code accompanying the paper "Massive Activations in Large Language Models"

[CVPR 2022] Deep Equilibrium Optical Flow Estimation

DC3: A Learning Method for Optimization with Hard Constraints

Modern Fixed Point Systems using Pytorch

Code for Finetune like you pretrain: Improved finetuning of zero-shot vision models

Learning perturbation sets for robust machine learning

Code release for "Idiosyncrasies in Large Language Models"

[ICML 2026] Code for Equilibrium Reasoners: learning attractor dynamics for scalable reasoning

The Limited Multi-Label Projection Layer

Enforcing robust control guarantees within neural network policies

Monotone operator equilibrium networks