Gymnax

https://github.com/roberttlange/gymnax

Gymnax provides reinforcement learning environments in JAX, leveraging JIT and vmap/pmap for...

Tokens:19,129
Snippets:82
Trust Score:9.6
License:Apache-2.0
Update:9 months ago
Tokens:
Raw