deepspeedai / deepspeedai/DeepSpeed
Fully half precision optimizer
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
Hi,
Is there an implementation of fully half-precision optimizer? I see from the GPT-3 paper:
... implemented an early version of the codebase, and developed the memory optimizations for fully
half-precision training.
It looks like DeepSpeed optimizer is still using mixed precision. Has anyone looked into this before? This would allow training these big models with much less memory ... Thanks!
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reviewing the DeepSpeed optimizer implementation and the GPT-3 paper passage cited in the issue. Determine the design needed for fully half-precision training, then verify that the resulting optimizer reduces memory use while preserving training behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100