search

RobertTLange / gymnax

RL Environments in JAX 🌍
Apache License 2.0
616 stars 61 forks source link

Automated tools for benchmarking #7

Closed RobertTLange closed 2 years ago

RobertTLange commented 3 years ago

I would like to have a set of utilities that automatically generates jobs and schedules them for the different hardware platforms. E.g. CPU/GPU/TPU. This should probably all run on GCP and with docker containers for reproducibility. Have a look pyhpc for benchmarking standards and potentially integrate with mle-toolbox GCP experiment setup.

RobertTLange commented 3 years ago

Also add nice visualizations!