Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: google/maxtext Loading
base: main@{1day}
Choose a base ref
...
head repository: google/maxtext Loading
compare: main
Choose a head ref
  • 2 commits
  • 7 files changed
  • 2 contributors

Commits on Aug 16, 2024

  1. Initial grad accumulate with scan

    Gradient accumulation
    
    manually tested
    
    globals are sus
    
    Add grad accumulation test
    
    Gradient accumulation config assert
    
    Gradient accumulation config assert
    
    pylint
    
    fixed tests
    
    global and micro batch size
    
    Clean up with microbatches
    
    Gradient accumulation
    
    Gradient accumulation
    
    Gradient accumulation
    
    Gradient accumulation
    
    Gradient accumulation
    
    Gradient accumulation
    
    Gradient accumulation
    
    grad acc
    
    grad acc
    gobbleturk committed Aug 16, 2024
    Configuration menu
    Copy the full SHA
    344b46c View commit details
    Browse the repository at this point in the history
  2. Merge pull request #812 from google:mattdavidow-grad-acc

    PiperOrigin-RevId: 663760161
    maxtext authors committed Aug 16, 2024
    Configuration menu
    Copy the full SHA
    ed21f6a View commit details
    Browse the repository at this point in the history
Loading