awslabs / awslabs/keys_values

Replace Lightning Fabric by elementary setup

Open
#135 2 comments 0 reactions 1 assignee Claimed by @mseeger View on GitHub
enhancement
Dominant language
Python
Stars
18
Forks
8
PR merge metrics
No merged PRs in 30d

Description

**Is your feature request related to a problem? Please describe.**
We've had many problems with Lightning Fabric doing thing behind the scenes. Right now, we only use it to spawn processes for the different ranks. This does not justify all its hidden complexity.

**Describe the solution you'd like**
Replace all Lightning Fabric dependencies by `torch.distributed` and `torch.multiprocessing` basics.

This could be done by asking Claude Code to do it.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.