Hi, I've updated brax and for some reason latest wheel on PyPI (brax==0.9.0) doesn't have torch wrapper module. It is present on the main here and in previous version v0.1.2
Hi @ikamensh thanks for using brax! It'll be pushed to PyPI once we bump the git tag – will update here once we do that! For now you can install from source
Hi, I've updated brax and for some reason latest wheel on PyPI (brax==0.9.0) doesn't have torch wrapper module. It is present on the main here and in previous version v0.1.2