Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)

verl: Volcano Engine Reinforcement Learning for LLMs

Official inference repo for FLUX.1 models

PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

Fitting latent circuit model to neural responses.

Implementing a ChatGPT-like LLM in PyTorch from scratch, step by step

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

PyTorch code and models for V-JEPA self-supervised learning from video.

MathVista: data, code, and evaluation for Mathematical Reasoning in Visual Contexts

Computational toolbox for large scale Calcium Imaging Analysis, including movie handling, motion correction, source extraction, spike deconvolution and result visualization.

A few CLI tools to immersively, browse, read Arxiv paper, save to notion, chat with them and save the chat history

Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series

A curated collection of neuroscience tasks with a common interface.

Trying out the Mamba architecture on small examples (cifar-10, shakespeare char level etc.)

Predictive auxiliary tasks in Deep RL as a model of hippocampal representation learning

Official implementation of DeepLabCut: Markerless pose estimation of user-defined features with deep learning for all animals incl. humans

Supplementary data and code accompanying: https://arxiv.org/abs/2109.12434

Tutorial for Harvard Medical School ML from Scratch Series: Transformer from Scratch. Demo the usage of transformer in various domains: Music sheet, audio signal, image generation & discrimination

A python package that includes many methods for decoding neural activity

Code and data to accompany the Rosenberg 2021 Preprint