LLM finetuning in resource-constrained environments.

Recipes for shrinking, optimizing, customizing cutting edge vision models. 💜

Simulation framework for accelerating research in Private Federated Learning

A Continual Learning Library in PyTorch and JAX

Implementing Vi(sion)T(transformer)

Benchmark for bi-level optimization solvers

A functional training loops library for JAX

Interactive LLM Powered NPCs, is an open-source project that completely transforms your interaction with non-player characters (NPCs) in any game! 🎮🤖🚀

Federated Learning with Partial Model Personalization

YOLOv8 TensorRT C++ Implementation

Code release for "Improved baselines for vision-language pre-training"

The M2L school 2022 tutorials

Dynamic Token Expansion with Continual Transformers, accepted at CVPR 2022

A crude RLHF layer on top of nanoGPT with Gumbel-Softmax trick

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

Neovim config for the lazy

Official PyTorch implementation of "Speeding up Heterogeneous Federated Learning with Sequentially Trained Superclients", accepted at ICPR 2022

Matplotlib styles for scientific plotting

A large labelled image dataset for benchmarking in federated learning

A little Python script to collect LaTeX sources for upload to the arXiv.

ASDL: Automatic Second-order Differentiation Library for PyTorch

Learning Across Domains and Devices: Style-Driven Source-Free Domain Adaptation in Clustered Federated Learning

Generalizing Federated Learning to Semantic Segmentation in Autonomous Driving

A curated list of awesome papers on dataset distillation and related applications.

Awesome Incremental Learning

A pure-functional implementation of a machine learning transformer model in Python/JAX

Official implementation of "Incremental Learning in Semantic Segmentation from Image Labels"