DLR-RM / stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
https://stable-baselines3.readthedocs.io
MIT License
8.99k stars 1.69k forks source link

[Feature Request] request title jax API and gymnax #2020

Open quanmissq opened 2 weeks ago

quanmissq commented 2 weeks ago

🚀 Feature

Suggest updating the stable version of the jax API and the algorithm API for gym

Motivation

No response

Pitch

No response

Alternatives

No response

Additional context

No response

Checklist

araffin commented 2 weeks ago

Hello, could you please provide more information.

Are you referring to https://github.com/DLR-RM/stable-baselines3/pull/1837 ?