Lightning-AI / Lightning-AI/litgpt

Support a new model

Open
#1,475 7 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

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

Description

Do you have a plan to support JetMoE model (https://github.com/myshell-ai/JetMoE) that very effective to reduce computational cost in inference in litgpt?

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

The issue names no litgpt files, tests, or entry points. Start by reviewing the JetMoE repository and how litgpt currently supports models to determine the integration scope. Done would require a defined implementation plan and completed JetMoE support, but the issue does not specify acceptance criteria.

Written by the indexing model from the issue text.

Assessment

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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.