Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

huggingface / transformers Public

Notifications You must be signed in to change notification settings
Fork 27.6k
Star 138k

Code
Issues 991
Pull requests 543
Actions
Projects 1
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Security
Insights

Flax whisper gradient checkpointing#22897

Closed

versae wants to merge 13 commits intohuggingface:mainhuggingface/transformers:mainfrom versae:flax-whisper-gradient-checkpointingversae/transformers:flax-whisper-gradient-checkpointingCopy head branch name to clipboard

+63-21

Conversation Commits13 (13)Checks Files changed

Commits

Commits on Apr 20, 2023

Adding gradient_checkpointing to Flax Whisper
versae
authored
Fixes issue with arguments
versae
authored

Commits on Apr 21, 2023

Update src/transformers/models/whisper/modeling_flax_whisper.py

versae
and
sanchit-gandhi
authored
Adding back kwargs to decoder_layer
versae
authored
Merge branch 'huggingface:main' into flax-whisper-gradient-checkpointing
versae
authored
Merge branch 'huggingface:main' into flax-whisper-gradient-checkpointing
versae
authored
Setting internal attr self.gradient_checkpointing
versae
authored
Merge branch 'huggingface:main' into flax-whisper-gradient-checkpointing
versae
authored

Commits on Apr 22, 2023

Adding gradient_checkpointing to self to it's available in setup()
versae
authored
Adding back proper kwargs
versae
authored
Removing kwargs to remat decoder layer
versae
authored
Black reformat
versae
committed
Fix check copies
versae
committed

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.