Skip to content

Guillem96/rl-examples-jax

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Reinforcement learning with Jax

Simple reinforcement learning examples with JAX and Haiku.

The aim of this repository is to provide simple examples on how to develop Deep Reinforcement learning using JAX.

References

  • JAX - Composable Transformations of Python+NumPy programs 2018

About

Simple reinforcement learning examples with JAX. https://github.com/google/jax

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages