Add an RL example in Jax

[lebrice]

No description provided.

[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 ⚠️
project/algorithms/jax_rl_example.py	93.02%	24 Missing ⚠️
project/trainers/jax_trainer.py	89.22%	18 Missing ⚠️
project/algorithms/callbacks/samples_per_second.py	41.66%	7 Missing ⚠️
project/main.py	0.00%	5 Missing ⚠️
project/conftest.py	78.94%	4 Missing ⚠️
project/utils/typing_utils/jax_typing_utils.py	90.90%	3 Missing ⚠️

Additional details and impacted files

@@            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     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.