Skip to content
New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

Speed up training by making use of jax.jit #2

Open
phinate opened this issue Mar 1, 2020 · 0 comments
Open

Speed up training by making use of jax.jit #2

phinate opened this issue Mar 1, 2020 · 0 comments
Assignees
@phinate
Labels
enhancement New feature or request

Comments

@phinate
Copy link
Member

phinate commented Mar 1, 2020

The working CLs training loop has so far not made use of the @jax.jit decorator. Doing so would dramatically speed up training time.
@phinate phinate added the enhancement New feature or request label Mar 1, 2020
@phinate phinate self-assigned this Mar 1, 2020
# for free to join this conversation on GitHub. Already have an account? # to comment
Assignees

@phinate phinate

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

No branches or pull requests
1 participant
@phinate