Add an RL example in Jax

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 86.33819% with 122 lines in your changes missing coverage. Please review.

Project coverage is 80.90%. Comparing base (cf197f8) to head (fac9cf2). Report is 4 commits behind head on master.

Files with missing lines	Patch %	Lines
project/algorithms/jax_rl_example_test.py	79.66%	61 Missing :warning:
project/algorithms/jax_rl_example.py	93.02%	24 Missing :warning:
project/trainers/jax_trainer.py	89.22%	18 Missing :warning:
project/algorithms/callbacks/samples_per_second.py	41.66%	7 Missing :warning:
project/main.py	0.00%	5 Missing :warning:
project/conftest.py	78.94%	4 Missing :warning:
project/utils/typing_utils/jax_typing_utils.py	90.90%	3 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #55 +/- ## ========================================== + Coverage 79.28% 80.90% +1.61% ========================================== Files 61 66 +5 Lines 3597 4457 +860 ========================================== + Hits 2852 3606 +754 - Misses 745 851 +106 ```

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