Add brax environments

Added brax environments, abstract solver, simulate, datatype handling and random key handling. CEM and MPC are adjusted so they can start from an external state.

Closes #20 (closed)