Stars
A curated list of awesome Machine Learning frameworks, libraries and software.
High-Resolution Image Synthesis with Latent Diffusion Models
A generative world for general-purpose robotics & embodied AI learning.
Graph Neural Network Library for PyTorch
PyTorch implementations of Generative Adversarial Networks.
End-to-End Object Detection with Transformers
Generate 3D objects conditioned on text or images
PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning
Pytorch implementation of convolutional neural network visualization techniques
[CVPR 2024] Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data. Foundation Model for Monocular Depth Estimation
Cosmos is a world model development platform that consists of world foundation models, tokenizers and video processing pipeline to accelerate the development of Physical AI at Robotics & AV labs. C…
Official repository of "SAMURAI: Adapting Segment Anything Model for Zero-Shot Visual Tracking with Motion-Aware Memory"
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.
[NeurIPS 2024] Depth Anything V2. A More Capable Foundation Model for Monocular Depth Estimation
Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet…
PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
Imitation learning algorithms with Co-training for Mobile ALOHA: ACT, Diffusion Policy, VINN
Unified framework for robot learning built on NVIDIA Isaac Sim
Papers and Datasets about Point Cloud.
Pytorch framework for doing deep learning on point clouds.
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
Isaac Gym Reinforcement Learning Environments
👀 Eye Tracking library easily implementable to your projects
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
Avalanche: an End-to-End Library for Continual Learning based on PyTorch.