Skip to content

Feat: Add Orchestration Layer for Evaluation #1640

Description

@ritaneves

Goal:

Gym needs a shared way to launch evaluation across teams, otherwise risking having a messy non-sharable launch/deployment setups.

Reasoning:

Our customers (e.g Nemotron) are used to the concept of launching jobs in Slurm. Gym assumes local execution, no layer that allows for execution on Slurm. Right now, benchmarks are being executed via custom, often cluster-specific sbatch scripts which are not OSSable.
Beyond this, supporting the concept of jobs allows for easier caching and running on the background.
The ultimate goal is to continue supporting this by having this layer orchestrated in Gym, with support for local and Kubernetes in the near the future.

Action Items:

Defining ownership.
Architecture discussion, bearing in mind we want to keep evaluations running in the simplest manner possible (i.e. not have a gym eval and a gym launcher eval, which will confuse people). Options for integration:
NeMo-Skills
Nemo-run
Nemo-RL
NEL-Next
New tool based on NEL-Next / NEL (focused on orchestration only) -> Gym Launcher

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

Status
Dev Todo

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions