Lightning-AI / Lightning-AI/litgpt

On the relationship with `torchtune`

Open
#1,309 1 comment 2 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
13.7k
Forks
1.5k
Avg merge
15h 37m
Merged PRs (30d)
1

Description

Will it be possible in the future for you to coordinate with the torchtune project so that we are able to use A for xyz and B for ikj?

We've been using litgpt a lot appreciating the nice balance between minimal modeling implementation and key quality of life features for distributed LLM training through the lightning ecosystem.

It seems like since torchtune, lightning, and litgpt are both at varying levels of torch nativeness and the tools for pretraining and finetuning do tend to diverge a bit, it would be cool for us as users of such toolkits if there's an effort to double down on this distinction so that we get optimal utility in all cases without heavy duplication of functionality across libraries.

I already see some nods to litgpt in the torchtune codebase but I wonder how intentional or open the channels of communication are here.

@carmocca
🤝
@rohan-varma

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No files, tests, or entry points are named. Start by reviewing the referenced torchtune project and the existing litgpt references there, then clarify the proposed division of functionality with the maintainers. Done would require an agreed, concrete coordination or integration scope rather than the current open-ended question.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ai, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.