HF RL Explorer

Evaluates the ability to implement pipeline parallel training for LLaMA using PyTorch distributed primitives…

Evaluates the ability to implement pipeline parallel training for LLaMA using PyTorch distributed primitives…: a task in terminal-bench-2-1-apptainer-v1: Terminal-Bench 2.1 (offline Apptainer, v1) (Harbor dataset). Implement pipeline parallel training for the LLaMA model using PyTorch. Create the…

The task

Implement pipeline parallel training for the LLaMA model using PyTorch. Create the file /app/pipeline_parallel.py and implement the following function according to the given signature:

Part of laion/terminal-bench-2-1-apptainer-v1.