Highlights
- Pro
Stars
PyTorch code for hierarchical k-means -- a data curation method for self-supervised learning
Claude Code Manager - The power-user toolkit for Claude Code
AI agents running research on single-GPU nanochat training automatically
State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!
A Simple and Universal Swarm Intelligence Engine, Predicting Anything. 简洁通用的群体智能引擎，预测万物
GitHub repository for the ICLR Computational Geometry & Topology Challenge 2021
AI agent toolkit: coding agent CLI, unified LLM API, TUI & web UI libraries, Slack bot, vLLM pods
NVIDIA FastGen: Fast Generation from Diffusion Models
A PyTorch implementation of "GeoSynth: Contextually-Aware High-Resolution Satellite Image Synthesis"
Fine-grained Image Editing by Pixel-wise Guidance Using Diffusion Models
Pointer Networks Implementation to solve Convex-Hull and TSP problems using supervised and RL training.
Lab Website
Lumina-mGPT 2.0: Stand-Alone AutoRegressive Image Modeling
Official PyTorch implementation for "Large Language Diffusion Models"
The image prompt adapter is designed to enable a pretrained text-to-image diffusion model to generate images with image prompt.
Training Sparse Autoencoders on Language Models
Deep Learning in Javascript. Train Convolutional Neural Networks (or ordinary ones) in your browser.
Influencer dataset collected from Instagram
A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
Simple and readable code for training and sampling from diffusion models
[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"
Repository for PURE: Turning Polysemantic Neurons Into Pure Features by Identifying Relevant Circuits, accepted at CVPR 2024 XAI4CV Workshop (spotlight)
[CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
New repo collection for NVIDIA Cosmos: https://github.com/nvidia-cosmos