Stars
Simulation benchmark from Toyota Research Institute containing 49 tasks that measure the performance of Large Behavior Model policies
Code to pretrain, fine-tune, and evaluate DreamZero and run sim & real-world evals
Enjoy the magic of Diffusion models!
A Practitioner's Guide to M(eow)ti Turn Agentic ReinfOrcement learning
Text Adventure Learning Environment Suite - Benchmark to evaluate language models on interactive text environments.
A PyTorch native platform for training generative AI models
Minimalistic 4D-parallelism distributed training framework for education purpose
LLM training code for Databricks foundation models
Implementation of paper Data Engineering for Scaling Language Models to 128K Context
A high-throughput and memory-efficient inference and serving engine for LLMs
🐟 Code and models for the NeurIPS 2023 paper "Generating Images with Multimodal Language Models".
Minimal implementation of Decision Transformer: Reinforcement Learning via Sequence Modeling in PyTorch for mujoco control tasks in OpenAI gym
Some preliminary explorations of Mamba's context scaling.
🔥 [CVPR2024] Official implementation of "Self-correcting LLM-controlled Diffusion Models (SLD)
Structured state space sequence models
LLM (Large Language Model) FineTuning
Modeling, training, eval, and inference code for OLMo
Official Code for the WWW'24 Paper: "Towards Explainable Harmful Meme Detection through Multimodal Debate between Large Language Models"
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
🙈 Code for Zero-shot Triplet Extraction by Template Infilling (Kim et al; IJCNLP-AACL 2023)
PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image
An open source implementation of CLIP.
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.