Slienteagle-wyb's repositories
hands_on_diffusion
some toy scripts based on book 'Hands-On Generative AI with Transformers and Diffusion Models'
agile_flight
Developing and Comparing Vision-based Algorithms for Vision-based Agile Flight
cleanrl
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, Ape-X DQN, DDPG, TD3, SAC)
COMPASS
COntrastive Multimodal Pretraining for AutonomouS Systems
ConvNeXt
Code release for ConvNeXt model
decision-transformer
Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.
DJITelloPy
DJI Tello drone python interface using the official Tello SDK. Feel free to contribute!
dreamerv2
Mastering Atari with Discrete World Models
flightmare
An Open Flexible Quadrotor Simulator
google-research
Google Research
gym
A toolkit for developing and comparing reinforcement learning algorithms.
gym-pybullet-drones
PyBullet Gym environments for single and multi-agent reinforcement learning of quadcopter control
iris
Transformers are Sample Efficient World Models
IsaacGymEnvs
Isaac Gym Reinforcement Learning Environments
myosuite
MyoSuite is a collection of environments/tasks to be solved by musculoskeletal models simulated with the MuJoCo physics engine and wrapped in the OpenAI gym API.
OmniIsaacGymEnvs
Reinforcement Learning Environments for Omniverse Isaac Gym
pydreamer
PyTorch implementation of DreamerV2 model-based RL algorithm
pytorch-grad-cam
Many Class Activation Map methods implemented in Pytorch for CNNs and Vision Transformers. Including Grad-CAM, Grad-CAM++, Score-CAM, Ablation-CAM and XGrad-CAM
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
quad-swarm-rl
Additional environments compatible with OpenAI gym
rlpyt
Reinforcement Learning in PyTorch
robomimic
robomimic: A Modular Framework for Robot Learning from Demonstration
safe-control-gym
PyBullet CartPole and Quadrotor environments—with CasADi symbolic a priori dynamics—for learning-based control and RL
sample-factory
High throughput synchronous and asynchronous reinforcement learning
solo-learn
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning
spr
Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"
tdmpc
Code for "Temporal Difference Learning for Model Predictive Control"