This is your work, valued
Deep Learning and Artificial Intelligence
DeepInfomaxPytorch. Learning deep representations by mutual information estimation and maximization
★ 321keypoints. Unsupervised learning of Object Landmarks through Conditional Image Generation
★ 49cma-es. Demo of Covariance Matrix Adaptation, Evolutionary Strategy
★ 22rnd. Exploration by Random Network Distillation
★ 15DualAttentionSeq2Seq. Analysis of Time Series data using Seq2Seq LSTM and 2 attention layers
★ 15mixturedensity. Mixture Density Network Demo in Pytorch
★ 14iic. Invariant Information Clustering for Unsupervised Image Classification and Segmentation
★ 10dreamerv3. Python
★ 9oardm. Autoregressive Diffusion Models
★ 3variationalcoders. Convolutional Autoencoder for compressing Atari images
★ 2APMFieldDeveloperSetup.
★ 2modern_robotics. examples from Modern Robotics by Lynch and Park
★ 2wandb-cleanup. Wandb model cleanup utility
★ 2crypto_resources.
★ 1maxpoolvsconv. Python
★ 1deep_rl. Deep RL demos
★ 1lightning_demo. pytorch lightning demo
★ 1deep_learning. deep learning template project
★ 1altara. React components for real-time telemetry dashboards — robotics, aerospace, autonomous vehicles, industrial IoT. PFD, HSI, time-series, gauges, GPS maps, and ROS2/MQTT adapters at 60fps.
★ 72gorilla. Gorilla: Training and Evaluating LLMs for Function Calls (Tool Calls)
★ 13kclaude-code-prompt-plunderer. 🔍 Extract and analyze your Claude Code conversation history to discover usage patterns, generate insights, and create custom slash commands from your prompting style.
★ 5dev-agent-lens. Proxy Claude Code Requests through LiteLLM to Arize
★ 68playwrightess-mcp. An experiment of using playwright MCP via JavaScript
★ 75clangd-tidy. A faster alternative to clang-tidy
★ 159nano-agent. A MCP Server for a small scale engineering agents with multi-provider LLM support.
★ 205mcp-gdb. JavaScript
★ 155nanobind. nanobind: tiny and efficient C++/Python bindings
★ 3.6kfastmcp. 🚀 The fast, Pythonic way to build MCP servers and clients.
★ 27kmadrona_mjx. C++
★ 163mujoco. Multi-Joint dynamics with Contact. A general purpose physics simulator.
★ 14kAnyNet. (ICRA) Anytime Stereo Image Depth Estimation on Mobile Devices
★ 521LWANet. (Neurocomputing) Light-weight network for real-time adaptive stereo depth estimation
★ 29PatchRefiner. [ECCV 2024] Leveraging Synthetic Data for Real-Domain High-Resolution Monocular Metric Depth Estimation
★ 68BiTTC. Python
★ 58Bi3D. Python
★ 173mujoco_warp. GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
★ 1.4kAlphaHero. Distributed AlphaZero implementation in Python.
★ 1pypop. [JMLR (CCF-A)] PyPop7: A Pure-PYthon LibrarY for POPulation-based Black-Box Optimization (BBO), especially *Large-Scale* algorithm variants. In the near future, we will add Learning-Based Optimizers as its extensions. [https://jmlr.org/papers/v25/23-0386.html]
★ 287Evolutionary-Optimization-Algorithms. DE, CMA-ES, MA-ES, LM-MAES Python Implementation
★ 31BARWindSpeed. Beyond All Reason game wind speed simulator
★ 1PyDnD. PyDnD is a Dungeons and Dragons SRD Library for Python3
★ 34dnd-character. library for making Dungeons & Dragons 5e characters as serializable data
★ 54open_spiel. OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
★ 5.4kauto-video-editing-suite. All the tools that allow me to never ever open up Final Cut
★ 11gymnax-blines. Baselines for gymnax 🤖
★ 78rejax. Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
★ 274evosax. Evolution Strategies in JAX 🦎
★ 781WU-UCT. A novel parallel UCT algorithm with linear speedup and negligible performance loss.
★ 124AlphaZero. Python
★ 36cestpasphoto.github.io. AlphaZero and other AI in your browser
★ 10alpha-zero-general. A very fast implementation of AlphaZero, applied to games like Splendor, Santorini, The Little Prince, … Browser version available
★ 66mctx. Monte Carlo tree search in JAX
★ 2.6kmctx-az. Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree
★ 27turbozero. fast + parallel AlphaZero in JAX
★ 112gymnax. RL Environments in JAX 🌍
★ 910pgx. ♟️ Vectorized RL game environments in JAX
★ 635turbozero_torch. fast + parallel AlphaZero in PyTorch
★ 15AgentBench. A Comprehensive Benchmark to Evaluate LLMs as Agents (ICLR'24)
★ 3.6kLightZero. [NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)
★ 1.6k1D-ARC.
★ 36Single-Player-MCTS. 🌳 Python implementation of single-player Monte-Carlo Tree Search.
★ 67torch-distmap. Euclidean distance transform in pytorch
★ 21LocNDF. LocNDF: Neural Distance Field Mapping for Robot Localization
★ 153maze-dataset. maze datasets for investigating OOD behavior of ML systems
★ 84cleanrl. High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
★ 10kAbsolute-Policy-Optimization. The source code of Absolute Policy Optimization
★ 7direct-preference-optimization. Reference implementation for DPO (Direct Preference Optimization)
★ 2.9krl. A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
★ 3.5ksource_one. Open Source FPV Drone Frame
★ 628IsaacLab. Unified framework for robot learning built on NVIDIA Isaac Sim
★ 7.8karuco_ros. Software package and ROS wrappers of the Aruco Augmented Reality marker detector library
★ 604master_project. This project is about vision based navigation and precision landing of a drone using ROS, PX4 and OpenCV
★ 70autonomous_landing_uav. ROS packages of the Autonomous landing system of a UAV in Gazebo
★ 214NvChad. Blazing fast Neovim framework providing solid defaults and a beautiful UI, enhancing your neovim experience.
★ 28kBattle-of-the-Backbones.
★ 212PointPillars. A Simple PointPillars PyTorch Implementation for 3D LiDAR(KITTI) Detection.
★ 888mmdetection3d. OpenMMLab's next-generation platform for general 3D object detection.
★ 6.5kPonderV2. [T-PAMI 2025] PonderV2: Pave the Way for 3D Foundation Model with A Universal Pre-training Paradigm
★ 376awesome-rgbd-datasets. This repository contains information for the paper "A Survey on RGB-D Datasets" and is a collaborative initiative to update the datasets list faster.
★ 2903D_Hand_Keypoint_Detection. Modified the InterHand2.6M for 3D hand detection on custom images for pose estimation using a Single RGB Image.
★ 15diffusion_policy. [RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion
★ 4.4kdiffusion-policy. Implementation of Diffusion Policy, Toyota Research's supposed breakthrough in leveraging DDPMs for learning policies for real-world Robotics
★ 136tapnet. Tracking Any Point (TAP)
★ 2kaot-benchmark. An efficient modular implementation of Associating Objects with Transformers for Video Object Segmentation in PyTorch
★ 595Segment-and-Track-Anything. An open-source project dedicated to tracking and segmenting any objects in videos, either automatically or interactively. The primary algorithms utilized include the Segment Anything Model (SAM) for key-frame segmentation and Associating Objects with Transformers (AOT) for efficient tracking and propagation purposes.
★ 3.1kCoDeF. [CVPR'24 Highlight] Official PyTorch implementation of CoDeF: Content Deformation Fields for Temporally Consistent Video Processing
★ 4.8kproduct-key-memory. Standalone Product Key Memory module in Pytorch - for augmenting Transformer models
★ 87mae. PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
★ 8.4ksegment_anything_tensorrt. Accelerate segment anything model inference using Tensorrt 8.6.1.6
★ 106segment-anything. The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
★ 55kx-transformers. A concise but complete full-attention transformer with a set of promising experimental features from various papers
★ 5.9kNYU-DLSP20. NYU Deep Learning Spring 2020
★ 6.8kact. Python
★ 2.1kfast-whisper-finetuning. Jupyter Notebook
★ 562Track-Anything. Track-Anything is a flexible and interactive tool for video object tracking and segmentation, based on Segment Anything, XMem, and E2FGVI.
★ 7kdinov2. PyTorch code and models for the DINOv2 self-supervised learning method.
★ 13kinterbotix_ros_manipulators. ROS Packages for Interbotix Arms
★ 182dreamerv3. Mastering Diverse Domains through World Models
★ 3.6kcrafter. Benchmarking the Spectrum of Agent Capabilities
★ 577solo-learn. solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
★ 1.6kadios. Shell
★ 64ROS-Mobile-Android. Visualization and controlling application for Android
★ 539roverrobotics_stack. ROS1 stack for Rover Pro, Rover Zero and Rover Zero 2. Its recommended all users of archived rr_openrover_stack update to this repo.
★ 9tensorrt_inference. C++
★ 718PaLM-rlhf-pytorch. Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM
★ 7.9kimutils. A series of convenience functions to make basic image processing operations such as translation, rotation, resizing, skeletonization, and displaying Matplotlib images easier with OpenCV and Python.
★ 4.6kSiamTrackers. (2020-2022)The PyTorch version of SiamFC,SiamRPN,DaSiamRPN, UpdateNet , SiamDW, SiamRPN++, SiamMask, SiamFC++, SiamCAR, SiamBAN, Ocean, LightTrack , TrTr, NanoTrack; Visual object tracking based on deep learning
★ 1.6kYOLOv7_Tensorrt. A simple implementation of Tensorrt YOLOv7
★ 109docker-mavros. Docker Container for MAVROS
★ 7fastbook. The fastai book, published as Jupyter Notebooks
★ 25kconnectbot. ConnectBot is the first SSH client for Android.
★ 3.4kcma-es. Covariance Matrix Adaptation Evolution Strategy (CMA-ES)
★ 82sutherland-hodgman. A differentiable Python implementation of the Sutherland–Hodgman algorithm for clipping polygons in 2D.
★ 50Python-Games. Python
★ 2Pygame-Geometry. My superset of pygame for making simulations and games.
★ 6zed-sdk. ⚡️The spatial perception framework for rapidly building smart robots and spaces
★ 1.2ksip-call. ESP32 simple VoIP phone based on SPH0645 mic and UDA1334A decoder
★ 160OpenKAI. OpenKAI: A modern framework for unmanned vehicle and robot control
★ 259prompt-to-prompt. Jupyter Notebook
★ 3.5kdeis. [ICLR 2023]DEIS: Fast Sampling of Diffusion Models with Exponential Integrator
★ 161bthelper. Bluetooth helper tools for setting up serial ports for e.g. SSH
★ 34yolo-tiling. Tile (Slice) YOLO Dataset for Small Objects Detection
★ 79RxCmd. Bluetooth control of Raspberry Pi
★ 1drive_ros_bb_to_obstacle. Converts BoundingBoxArray.msg to ObstacleArray.msg using the camera homography.
★ 1gstream_Deep. ZED2 with Nvidia Deepstream
★ 7deepsparse. Sparsity-aware deep learning inference runtime for CPUs
★ 3.2krobo_magellan. Python
★ 4coco-annotator. :pencil2: Web-based image segmentation tool for object detection, localization, and keypoints
★ 2.3kcartographer. Cartographer is a system that provides real-time simultaneous localization and mapping (SLAM) in 2D and 3D across multiple platforms and sensor configurations.
★ 7.9kdeep-video-mvs. Code for "DeepVideoMVS: Multi-View Stereo on Video with Recurrent Spatio-Temporal Fusion" (CVPR 2021)
★ 233tsdf-fusion-python. Python code to fuse multiple RGB-D images into a TSDF voxel volume.
★ 1.4kpyRANSAC-3D. A python tool for fitting primitives 3D shapes in point clouds using RANSAC algorithm
★ 663RGBD-semantic-segmentation. A paper list of RGBD semantic segmentation (processing)
★ 421torchdyn. A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
★ 1.6kminGPT. A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
★ 25kfunctorch. functorch is JAX-like composable function transforms for PyTorch.
★ 1.4kELFIO. ELFIO and ARIO are robust, header-only C++ libraries designed to make binary file and archive manipulation effortless, efficient, and portable. Whether you are building compilers, linkers, binary analysis tools, or custom build systems, these libraries provide the foundation you need for working with ELF files and UNIX archives.
★ 885iris. Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.
★ 898DeepStream-Yolo. NVIDIA DeepStream SDK 8.0 / 7.1 / 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
★ 2kyolov7. Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
★ 14kyolo-tensorrt. TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
★ 1.2kpysot. SenseTime Research platform for single object tracking, implementing algorithms like SiamRPN and SiamMask.
★ 4.6kSiamTracker-with-TensorRT. The modify PyTorch version of Siam-trackers which are speed-up by TensorRT. [Updating...]
★ 19SiamMask. [CVPR19/TPAMI23] SiamMask: A Framework for Fast Online Object Tracking and Segmentation
★ 3.6kgoturn-pytorch. GOTURN tracking implemented in PyTorch
★ 85GOTURN. Source code for paper: Learning to Track at 100 FPS with Deep Regression Networks, Held, et al. ECCV 2016
★ 897jetson-nano-baseboard. Antmicro's open hardware baseboard for the NVIDIA Jetson Nano, TX2 NX and Xavier NX
★ 533MinAtar. Python
★ 333mmap.ninja. Memory mapped numpy arrays of varying shapes
★ 310mediamtx. Ready-to-use Media-over-QUIC / SRT / WebRTC / RTSP / RTMP / LL-HLS / MPEG-TS / RTP live media server and media proxy that allows to read, publish, proxy, record and playback real-time video and audio streams.
★ 20knetron. Visualizer for neural network, deep learning and machine learning models
★ 33kperceiver-pytorch. Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
★ 1.2kCFPNet-Medicine. CFPNet-M: A Light-Weight Encoder-Decoder Based Network for Multimodal Biomedical Image Real-Time Segmentation
★ 227gradio. Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!
★ 43kSparseInst. [CVPR 2022] SparseInst: Sparse Instance Activation for Real-Time Instance Segmentation
★ 621PyTorch-ONNX-TRT. TensortRT installation and Conversion from PyTorch Models
★ 36mobilestereonet. Lightweight stereo matching network based on MobileNet blocks
★ 310ONNX-MobileStereoNet. Python scripts for performing stereo depth estimation using the MobileStereoNet model in ONNX
★ 50docker-jetson-qemu. Shell
★ 20jetson-containers. Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
★ 4.8kjetson_stats. 📊 Simple package for monitoring and control your NVIDIA Jetson [Orin, Xavier, Nano, TX] series
★ 2.6kjetson-image. 💾 Create minimalist, Ubuntu based images for the Nvidia jetson boards
★ 578unet_plus_attention. Unet model with stand-alone self-attention blocks
★ 9segmentation_models.pytorch. Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
★ 12kSwin-Transformer. This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
★ 16kSeqFormer. SeqFormer: Sequential Transformer for Video Instance Segmentation (ECCV 2022 Oral)
★ 350UNet-family. Paper and implementation of UNet-related model.
★ 2.6kurl_benchmark. Python
★ 368pixel_models. Pytorch implementations of autoregressive pixel models - PixelCNN, PixelCNN++, PixelSNAIL
★ 36ladder-vae-pytorch. Ladder Variational Autoencoders (LVAE) in PyTorch
★ 90deformer. An order-agnostic distribution estimating Transformer.
★ 8denoising_diffusion. Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)
★ 91SDEdit. PyTorch implementation for SDEdit: Image Synthesis and Editing with Stochastic Differential Equations
★ 1.2kscore-denoise. :snowflake: Score-Based Point Cloud Denoising (ICCV 2021)
★ 215jax-variational-diffwave. Jax/Flax implementation of Variational-DiffWave.
★ 40pytorch_geometric. Graph Neural Network Library for PyTorch
★ 24kNeuralRecon. Code for "NeuralRecon: Real-Time Coherent 3D Reconstruction from Monocular Video", CVPR 2021 oral
★ 2.3kgradsim. Differentiable simulation for system identification and visuomotor control
★ 221jaxlie. Rigid transforms + Lie groups for JAX
★ 337liegroups. Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch
★ 359OpenDUNE. DUNE, REINVENTED
★ 1.7kcarla-visual-odometry. Visual odometry using stereo camera with CARLA simulator
★ 21spconv. Spatial Sparse Convolution Library
★ 2.3ksgnn. [CVPR'20] SG-NN: Sparse Generative Neural Networks for Self-Supervised Scene Completion of RGB-D Scans
★ 191Atlas. Atlas: End-to-End 3D Scene Reconstruction from Posed Images
★ 1.9kCPG-RBFN-framework. Code availability for M. Thor et al. 2021 submitted to NMI
★ 24RAFT-Stereo. Python
★ 1.1kHTM-pytorch. Implementation of Hierarchical Transformer Memory (HTM) for Pytorch
★ 74brax. Massively parallel rigidbody physics simulation on accelerator hardware.
★ 3.2kmicrograd. A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
★ 17kmbpo. Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"
★ 558Realsense-disparity-refinement-with-CUDA. C
★ 23lightning-kitti. Semantic Segmentation with Pytorch-Lightning
★ 63RealtimeStereo. Attention-Aware Feature Aggregation for Real-time Stereo Matching on Edge Devices (ACCV, 2020)
★ 185PSMNet. Pyramid Stereo Matching Network (CVPR2018)
★ 1.5kGwcNet. Group-wise Correlation Stereo Network, CVPR 2019
★ 338PyTorch-HITNet-Hierarchical-Iterative-Tile-Refinement-Network-for-Real-time-Stereo-Matching. HITNet implementation using PyTorch
★ 138PENet_ICRA2021. ICRA 2021 "Towards Precise and Efficient Image Guided Depth Completion"
★ 361X-StereoLab. SOS IROS 2018 GOOGLE; StereoNet ECCV2018 GOOGLE; ActiveStereoNet ECCV2018 Oral GOOGLE; HITNET CVPR2021 GOOGLE;PLUME Uber ATG
★ 723YOLOP. You Only Look Once for Panopitic Driving Perception.(MIR2022)
★ 2.2kDROID-SLAM. Python
★ 2.6kfast-depth. ICRA 2019 "FastDepth: Fast Monocular Depth Estimation on Embedded Systems"
★ 1kExternal-Attention-pytorch. 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
★ 12kjaxopt. Hardware accelerated, batchable and differentiable optimizers in JAX.
★ 1.1kORB_SLAM3. ORB-SLAM3: An Accurate Open-Source Library for Visual, Visual-Inertial and Multi-Map SLAM
★ 8.9kBayesian_Hilbert_Maps. Mapping spatiotemporal patterns in an online and continuous fashion
★ 111tinyloader. Python
★ 71pytube. Lightweight, dependency-free Python library and CLI for downloading YouTube videos, playlists, and captions.
★ 13kalbumentations. Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
★ 15kpytorch-image-models. The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
★ 37kvit-pytorch. Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
★ 25ktransfuser. [PAMI'23] TransFuser: Imitation with Transformer-Based Sensor Fusion for Autonomous Driving; [CVPR'21] Multi-Modal Fusion Transformer for End-to-End Autonomous Driving
★ 1.6ka0lite. A neural net chess engine in 95 lines of python
★ 76batch_rl. Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games
★ 560lightning-bolts. Toolbox of models, callbacks, and datasets for AI/ML researchers.
★ 1.8kCQL. Code for conservative Q-learning
★ 486taichi_three. A soft renderer based on Taichi (work in progress)
★ 232pyrigidbody3d. tiny toy rigid body sim in python
★ 34