Skip to content

Instantly share code, notes, and snippets.

@junpenglao

junpenglao/theano-jax-test-drive.ipynb

Last active November 10, 2020 07:46
Show Gist options
  • Save junpenglao/fe5e1b451c076cc7b4ca16acdd7d6472 to your computer and use it in GitHub Desktop.
Save junpenglao/fe5e1b451c076cc7b4ca16acdd7d6472 to your computer and use it in GitHub Desktop.
Download ZIP
theano-jax test drive.ipynb
Display the source blob
Display the rendered blob
Raw
theano-jax-test-drive.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@twiecki
Copy link

twiecki commented Sep 27, 2020

@junpenglao would be interesting to try with the pymc3 grad function rather than the jax-derived one.

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