Skip to content

Gradient accumulation with single graph and lax.scan#6

Open
apoorvtintin wants to merge 9 commits intomainfrom
gradient_accum_scan_clean
Open

Gradient accumulation with single graph and lax.scan#6
apoorvtintin wants to merge 9 commits intomainfrom
gradient_accum_scan_clean

Conversation

@apoorvtintin
Copy link
Copy Markdown
Collaborator

@apoorvtintin apoorvtintin commented Apr 11, 2024

Change GA configuration from fuji.py

@apoorvtintin apoorvtintin changed the title Gradient accum scan clean Gradient accumulation with single graph and lax.scan Apr 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant