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

Add training loop fusion #101

Open
seanmor5 opened this issue Aug 27, 2021 · 0 comments
Open

Add training loop fusion #101

seanmor5 opened this issue Aug 27, 2021 · 0 comments
Labels
area:loop Applies to loop API kind:feature New feature or request

Comments

@seanmor5
Copy link
Contributor

With the support of while loops in Nx and progress towards streaming data to the device, we should consider an API which allows for fusing the training loop into a single XLA computation. We will have to consider how this ties in with callbacks, hooks, etc.
@seanmor5 seanmor5 added kind:feature New feature or request area:loop Applies to loop API labels Aug 27, 2021
@seanmor5 seanmor5 mentioned this issue Oct 12, 2021
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
area:loop Applies to loop API kind:feature New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@seanmor5