Stars
My solutions to the problems in Fifty Challenging Problems in Probability by Frederick Mosteller
A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes practical examples for both text and image modalities.
Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI
Official implementation of the paper: "FlowEdit: Inversion-Free Text-Based Editing Using Pre-Trained Flow Models"
Best practices & guides on how to write distributed pytorch training code
Instructions for setting up a SLURM cluster using Ubuntu 18.04.3 with GPUs.
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
Yet another Hydar Torchrun Launcher
This is a proof of concept on how MCTS can be implemented on top of TorchRL
Codebase for Prioritizing samples in Reinforcement Learning with Reducible Loss
Causal discovery algorithms and tools for implementing new ones
RoMa: A lightweight library to deal with 3D rotations in PyTorch.
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.
Easily train AlphaZero-like agents on any environment you want!
VF3 Algorithm - The fastest algorithm to solve subgraph isomorphism on large and dense graphs
🦜🔗 Build context-aware reasoning applications
The Startup CTO's Handbook, a book covering leadership, management and technical topics for leaders of software engineering teams
For optimization algorithm research and development.
This repo contains annotated research papers that I found really good and useful
List of papers studying machine learning through the lens of category theory
A comprehensive guide to building RAG-based LLM applications for production.