deepspeedai / deepspeedai/DeepSpeed
Can DeepSpeed support Keras models based on the Torch backend?[REQUEST]
Open
Nobody has claimed this yet.
enhancement
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
Keras successfully supports Torch as its backend in version 3.0. Can I use DeepSpeed to train this model?
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
The issue mentions Keras 3.0 with its Torch backend and DeepSpeed, but names no files, tests, or entry points. Start by locating the integration boundaries for Keras, PyTorch, and DeepSpeed, then determine the supported behavior and validation needed for a Keras model using the Torch backend.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- keras, python, pytorch
- Domain
- distributed-systems, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 15/100