Skip to content

@locuslab CMU Locus Lab

Change the repository type filter

All

    Repositories list

    72 repositories

    • open-unlearning

      Public
      [NeurIPS D&B '25] The one-stop repository for large language model (LLM) unlearning. Supports TOFU, MUSE, WMDP, and many unlearning methods with easy feature extensibility.
      open-sourcebenchmarksright-to-be-forgottenprivacy-protectionunlearningmembership-inference-attacksmembership-inferencellmsllm-privacyllm-unlearning
      Python
      •
      MIT License
      •118•447•11•5•Updated Dec 5, 2025Dec 5, 2025

    • antidistillation-sampling

      Public
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •2•15•2•0•Updated Nov 4, 2025Nov 4, 2025

    • safety-pretraining

      Public
      HTML
      •0•3•0•0•Updated Sep 16, 2025Sep 16, 2025

    • confidential_llms

      Public
      Repository for the paper "Evaluating Language Model Reasoning about Confidential Information" and the PasswordEval benchmark.
      Python
      •0•1•0•0•Updated Aug 28, 2025Aug 28, 2025

    • llm-idiosyncrasies

      Public
      Code release for "Idiosyncrasies in Large Language Models"
      Python
      •
      MIT License
      •7•50•1•0•Updated Jul 21, 2025Jul 21, 2025

    • tofu

      Public
      Python
      •
      MIT License
      •3•20•2•0•Updated Feb 25, 2025Feb 25, 2025

    • acr-memorization

      Public
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •8•37•0•0•Updated Dec 19, 2024Dec 19, 2024

    • edge-of-stability

      Public
      Python
      •24•73•1•0•Updated Dec 7, 2024Dec 7, 2024

    • context-parametric-inversion

      Public
      1•4•2•0•Updated Nov 17, 2024Nov 17, 2024

    • llava-token-compression

      Public
      Python
      •3•46•0•0•Updated Nov 8, 2024Nov 8, 2024

    • ect

      Public
      Consistency Models Made Easy
      Python
      •12•321•9•1•Updated Oct 13, 2024Oct 13, 2024

    • diffusion-model-hallucination

      Public
      Python
      •
      MIT License
      •6•49•1•0•Updated Sep 5, 2024Sep 5, 2024

    • qpth

      Public
      A fast and differentiable QP solver for PyTorch.
      machine-learningdeep-learningoptimizationpytorchquadratic-programming
      Python
      •
      Apache License 2.0
      •111•773•15•0•Updated Sep 3, 2024Sep 3, 2024

    • mpc.pytorch

      Public
      A fast and differentiable model predictive control (MPC) solver for PyTorch.
      Python
      •
      MIT License
      •161•1k•18•0•Updated Aug 27, 2024Aug 27, 2024

    • wanda

      Public
      A simple and effective LLM pruning approach.
      network-pruninglarge-language-models
      Python
      •
      MIT License
      •121•825•45•1•Updated Aug 9, 2024Aug 9, 2024

    • fast_adversarial

      Public
      [ICLR 2020] A repository for extremely fast adversarial training using FGSM
      Python
      •89•450•3•2•Updated Jul 25, 2024Jul 25, 2024

    • projected_sinkhorn

      Public
      Python
      •15•88•1•1•Updated Jul 25, 2024Jul 25, 2024

    • robust_union

      Public
      [ICML'20] Multi Steepest Descent (MSD) for robustness against the union of multiple perturbation models.
      Python
      •4•25•0•1•Updated Jul 25, 2024Jul 25, 2024

    • lml

      Public
      The Limited Multi-Label Projection Layer
      Python
      •
      MIT License
      •6•59•0•1•Updated Jul 25, 2024Jul 25, 2024

    • scaling_laws_data_filtering

      Public
      Python
      •4•65•0•0•Updated Apr 9, 2024Apr 9, 2024

    • massive-activations

      Public
      Code accompanying the paper "Massive Activations in Large Language Models"
      large-language-models
      Python
      •
      MIT License
      •13•187•4•1•Updated Mar 4, 2024Mar 4, 2024

    • chatllm-vscode

      Public
      TypeScript
      •
      MIT License
      •7•62•2•0•Updated Feb 4, 2024Feb 4, 2024

    • get

      Public
      Generative Equilibrium Transformer
      Python
      •
      MIT License
      •2•26•0•0•Updated Nov 11, 2023Nov 11, 2023

    • torchdeq

      Public
      Modern Fixed Point Systems using Pytorch
      fixed-pointdeqequilibriumfixed-point-method
      Python
      •
      MIT License
      •13•125•9•0•Updated Oct 31, 2023Oct 31, 2023

    • T-MARS

      Public
      Code for T-MARS data filtering
      Python
      •
      MIT License
      •5•35•1•0•Updated Aug 23, 2023Aug 23, 2023

    • FLYP

      Public
      Code for Finetune like you pretrain: Improved finetuning of zero-shot vision models
      Python
      •
      MIT License
      •14•104•6•0•Updated Aug 13, 2023Aug 13, 2023

    • smoothinv

      Public
      Single Image Backdoor Inversion via Robust Smoothed Classifiers
      Python
      •
      MIT License
      •1•17•1•0•Updated Jul 18, 2023Jul 18, 2023

    • tta_conjugate

      Public
      Test-Time Adaptation via Conjugate Pseudo-Labels
      Python
      •
      MIT License
      •3•43•1•0•Updated May 25, 2023May 25, 2023

    • MonotoneDBM

      Public
      Python
      •1•2•0•0•Updated Apr 7, 2023Apr 7, 2023

    • cfd-gcn

      Public
      Python
      •36•127•4•1•Updated Mar 28, 2023Mar 28, 2023