google-deepmind / google-deepmind/mujoco_playground
jax-->ONNX
Open
- Dominant language
- Python
- Stars
- 2.2k
- Forks
- 359
- Avg merge
- 7d 3h
- Merged PRs (30d)
- 1
Description
Hello, author. While using brax_network_to_onnx.ipynb, I encountered an issue: Brax requires ml-dtypes version greater than 0.5.0, but the latest version of TensorFlow seems to not support such a high version of ml-dtypes. I would like to ask how you resolved this version conflict to successfully complete the model conversion from JAX to ONNX?
Contributor guide
Assessment
This issue has not been assessed yet.