pytorch / rl

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
https://pytorch.org/rl
MIT License
2.32k stars 307 forks source link

[Feature Request] Unity MLAgents Support #1110

Closed hyerra closed 1 month ago

hyerra commented 1 year ago

Motivation

Unity is a popular method for creating game environments for RL algorithms. I was wondering if potentially adding support for Unity's MLAgents framework could be considered. Here is some documentation to their low level API that could be useful for supporting Unity games:

https://github.com/Unity-Technologies/ml-agents/blob/develop/docs/Python-LLAPI.md https://github.com/Unity-Technologies/ml-agents/blob/develop/docs/Python-LLAPI-Documentation.md

Solution

Adding direct support for Unity MLAgent environments.

Checklist

hyerra commented 1 year ago

Just wanted to follow up on this. I could help with this feature request if this is something we wanted to pursue.