Add Classic Control Gymnax

[benjamc]

No description provided.

[benjamc]

Pause until new CARL env was integrated.

[TheEimer]

Why call them Jax and not Gymnax? I bet there are other attempts at Jaxing them, Gymnax would be more explicit imo

[TheEimer]

Env params is frozen for a reason, but I guess this runs, right? Pretty sure we can't jit it, but we have this issue with brax as well... at some point we might want to brainstorm if we can somehown avoid replacing env parts to change context. For now it's alright, though.

[TheEimer]

Looks good! One question though: have you tested if they're actually faster? In brax our context changes kind of break the speed, does it also happen here?

[codecov]

Codecov Report

Merging #90 (289d324) into development (f30d834) will increase coverage by 29.05%.
Report is 796 commits behind head on development.
The diff coverage is 49.43%.

❗ Current head 289d324 differs from pull request most recent head 7c2e21e. Consider uploading reports for the commit 7c2e21e to get more accurate results

Additional details and impacted files

@@               Coverage Diff                @@
##           development      #90       +/-   ##
================================================
+ Coverage        19.05%   48.10%   +29.05%     
================================================
  Files               78       84        +6     
  Lines             4030     4122       +92     
  Branches           555      543       -12     
================================================
+ Hits               768     1983     +1215     
+ Misses            3224     2060     -1164     
- Partials            38       79       +41     

[benjamc]

• Jaxify CARLEnv:

CARLJaxEnv:

• env state: sub env state + context id
• env params: empty
• gymnax environment api

[amsks]

@benjamc Is this still feasible to integrate into the main package? And do you have an estimate of how much work this is? If not, then we could potentially remove this branch