Add an RL example in Jax

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 77.24138% with 198 lines in your changes missing coverage. Please review.

Project coverage is 72.30%. Comparing base (52c88ce) to head (d87828c). Report is 3 commits behind head on master.

Files with missing lines	Patch %	Lines
project/algorithms/jax_rl_example_test.py	54.32%	132 Missing :warning:
project/algorithms/jax_rl_example.py	92.44%	26 Missing :warning:
project/algorithms/jax_trainer.py	89.15%	18 Missing :warning:
project/algorithms/callbacks/samples_per_second.py	41.66%	7 Missing :warning:
project/utils/typing_utils/jax_typing_utils.py	77.41%	7 Missing :warning:
project/main.py	0.00%	5 Missing :warning:
project/conftest.py	78.57%	3 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #55 +/- ## ========================================== + Coverage 70.99% 72.30% +1.31% ========================================== Files 59 63 +4 Lines 3551 4395 +844 ========================================== + Hits 2521 3178 +657 - Misses 1030 1217 +187 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.