Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Try the new JAX functionality in Theano #23

Open
brandonwillard opened this issue Oct 1, 2020 · 1 comment
Open

Try the new JAX functionality in Theano #23

brandonwillard opened this issue Oct 1, 2020 · 1 comment
Assignees
@rlouf
Labels
enhancement New feature or request

Comments

@brandonwillard
Copy link
Contributor

brandonwillard commented Oct 1, 2020 •
edited by sync-by-unito bot
Loading

We should see if our implementations can be converted to JAX, and whether or not that offers any performance advantages.

We start by attempting conversion (e.g. using jax_funcify) of our custom Distributions and, from there, find out if there are any Op conversions that are currently unimplemented. If that's the case, we write the missing conversions.
@brandonwillard brandonwillard added the enhancement New feature or request label Oct 1, 2020
@rlouf
Copy link

rlouf commented Mar 13, 2021 •
edited
Loading

I'll take this up.

@rlouf rlouf self-assigned this Mar 18, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@rlouf rlouf

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@brandonwillard @rlouf