NVIDIA-NeMo / NVIDIA-NeMo/RL

Long sequence length (1M+ seq len) memory efficiency

Open
#885 1 comment 0 reactions 0 assignees View on GitHub
research t-mcore
Dominant language
Python
Stars
2k
Forks
562
Avg merge
4d 5h
Merged PRs (30d)
145

Description

We would like to be able to train models up to 1M seq len. The number of nodes necessary should also be reasonable. We are ok if the training runs take more time i.e. no runtime efficiency constraints, we just want to be able to do it.

Contributor guide

Open the contributing guide

Research direction

The issue names no files, tests, or entry points. Start by locating the training path that controls sequence length and measuring memory and node requirements at large lengths. Done should demonstrate training at 1M+ sequence length with a reasonable node count, without requiring runtime efficiency.

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
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.