jetson-reinforcement

Reinforcement learning library

Provides tools and libraries for training intelligent agents to perform complex tasks in robotics using deep reinforcement learning and GPU acceleration.

Deep reinforcement learning GPU libraries for NVIDIA Jetson TX1/TX2 with PyTorch, OpenAI Gym, and Gazebo robotics simulator.

GitHub

884 stars
62 watching
224 forks
Language: C++
last commit: almost 3 years ago

Related projects:

Repository Description Stars
dusty-nv/ros_deep_learning A set of ROS/ROS2 nodes for deep learning inference on NVIDIA devices 901
google-deepmind/spriteworld An environment designed to test and train reinforcement learning algorithms in a flexible, procedurally generated 2D space with various objects and interactions. 368
hubfire/muti-branch-ddpg-carla An implementation of a reinforcement learning algorithm using multi-branch architecture and Deep Deterministic Policy Gradients (DDPG) to control autonomous vehicles in simulation environments. 79
jhu-lcsr/good_robot Research on repurposing reinforcement learning for transfer between tasks in robotics and multi-step visual tasks with simulation-to-real transfer 107
nicrusso7/rex-gym Training an open-source quadruped robot to perform domestic and generic tasks using reinforcement learning and OpenAI Gym. 988
resibots/chatzilygeroudis_2018_rte Research on an algorithm to learn how robots recover from damage in simulations and real-world environments 10
iffix/machin An open-source reinforcement learning library for PyTorch, providing a simple and clear implementation of various algorithms. 401
coreylowman/dfdx A deep learning library for Rust with GPU acceleration and ergonomic API. 1,737
toni-sm/skrl A modular reinforcement learning library with support for various environments and frameworks 560
nathanepstein/pavlov.js Reinforcement learning using Markov Decision Processes in C++ compiled to JavaScript 500
luchris429/purejaxrl A high-performance implementation of reinforcement learning training pipelines using JAX and PyTorch-like functionality 722
nadavbh12/retro-learning-environment A framework for training AI algorithms using game consoles as input 185
matyama/deep-rl-hands-on A collection of examples and exercises from a deep reinforcement learning book 2
tristandeleu/pytorch-maml-rl Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks 827
alex-petrenko/sample-factory A high-throughput reinforcement learning library with optimized synchronous and asynchronous implementations of policy gradients. 830