Sirsirious/change_backend.py

Created December 14, 2020 17:37

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/Sirsirious/2964a00a9c632401b9fa8c892487b0b8.js"></script>
Save Sirsirious/2964a00a9c632401b9fa8c892487b0b8 to your computer and use it in GitHub Desktop.

Download ZIP

Raw

change_backend.py

trax.fastmath.set_backend("tensorflow-numpy")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment