- As titled
- Solved: #149 

Waiting for testing from @chujiezheng

---------

Co-authored-by: Chi Zhang <zhangchi.usc1992@bytedance.com>

- Add link to performance tuning

- Previous gradient accumulation value is computed by micro_batch_size,
which is wrong when using dynamic_bsz
- Fix ci script to avoid overlooking this issue
- Change vLLM state log default value to True to disable log.
- We will check the `self.config.actor.ppo_mini_batch_size %
self.config.actor.ppo_micro_batch_size_per_gpu == 0` after normalization
in fsdp_workers instead of in dp_actor and dp_critic.

- As titled

# Add Sequence Parallelism and Padding Removal to SFT Trainer

This PR adds sequence parallelism (SP) and padding removal optimizations
to the SFT trainer, which can help improve training efficiency for large
language models.

## Key Changes

### Core Features
1. **Sequence Parallelism**: Added support for sequence parallelism
through the Ulysses framework
   - Configurable via `ulysses_sequence_parallel_size` parameter
   - Properly handles data distribution across SP ranks
   - Maintains consistent loss computation across distributed setup

2. **Padding Removal**: Added support for efficient handling of
variable-length sequences
   - Enabled via `use_remove_padding` flag (requires SP to be enabled)
   - Uses flash-attention's padding removal utilities
   - Handles proper re-padding and loss computation

3. **Training Improvements**:
   - Added label smoothing support to loss computation
   - Added progress bar with epoch information
   - Added RoPE scaling configuration support
   - Improved error messages for batch size validation

### Testing
- Added comprehensive test suite (`test_trainer.py`) to verify:
- Forward pass consistency between original and SP+rmpad implementations
  - Loss computation correctness across distributed setup
  - Proper handling of micro-batches

### Example Usage
Added example script `examples/sft/gsm8k/run_qwen_05_sp2.sh`
demonstrating how to use the new features with Qwen-2.5B model.

## Implementation Details
- Uses device mesh for proper distributed training setup
- Handles data distribution ensuring same sequences within SP groups but
different across DP groups
- Carefully manages backward pass timing with gradient checkpointing
- Maintains compatibility with existing FSDP features

## Testing Instructions
1. Run the example script with sequence parallelism:
```bash
bash examples/sft/gsm8k/run_qwen_05_sp2.sh <nproc_per_node> <save_path>
```

2. Run the test suite:
```bash tests/sft/run_sft_sp_loss_match.sh```


^^ These are PR description generated by [OpenHands](https://github.com/All-Hands-AI/OpenHands)

---------

Co-authored-by: Jiayi Pan <i@jiayipan.me>
Co-authored-by: openhands <openhands@all-hands.dev>

We set `max_num_batched_tokens` in config `.rollout`, but they weren't
actually being passed to VLLM -- causing potential insufficient use of
GPUs.

This PR:

- properly pass `max_num_batched_tokens` from config to vLLM
- set `disable_log_stats` to False, so vLLM performance information can
be properly displayed (to spot issues)

## Summary

This PR changes all the micro_batch_size to micro_batch_size_per_gpu.

**The Core logic of setting batch size:**
- **All algorithmic metrics** (train batch size, ppo mini batch size):
are global (from the perspective of single-controller), which will be
normalized in each Worker.
- **All performance-related parameters** (micro batch size, max token
length in dynamic batch size) are local parameters, which represent the
data sizes per GPU (i.e., each Worker).

## Main Changes

1. Change the scripts and config and delete the normalization for
micro_bsz
2. Fix CI for SFT

- As titled

minor fix

This PR adds support for LoRA (Low-Rank Adaptation) for efficient model
fine-tuning.

### Changes

1. Added LoRA configuration support in trainer config
2. Modified FSDP wrapping policy to handle LoRA modules
3. Integrated with existing FSDP training infrastructure
4. Added peft dependency
5. Removed unused ring_attn_utils.py

### Features

- Configurable LoRA rank and alpha parameters
- Target module specification for selective adaptation
- Compatible with FSDP sharding strategy

### Testing

Tested with Qwen2.5-0.5B-Instruct model on GSM8K dataset using the
provided example script.

### Dependencies

- Added `peft` package to requirements.txt

This PR is based on commit 902ddbe6 and has been merged with the latest
upstream main branch.

---------

Co-authored-by: Jiayi Pan <i@jiayipan.me>
Co-authored-by: openhands <openhands@all-hands.dev>

- Support training several iters in SFT trainer
- Add CI for SFT trainer to train one iter.

- Force ref/rm to use CPUOffload. Fix root FSDP unit not reshard weights
after forward
- HSDP support is on hold and assert False right now.

This reverts commit 19840945.

This PR supports:
- meta device init (which keeps the shared parameters)
- parallel pre-trained weight init for FSDP from huggingface checkpoint

---------

Co-authored-by: zhiqi.0 <zhiqi.0@bytedance.com>

- Implement KL loss, GRPO outcome adv, and utilize bon rollouts
- Provide scripts for deepseek and qwen on GSM8k. Can provide more for
other datasets.
- Support seq balance
- Train using qwen2-7b, GSM8k score can reach 0.89

- The actual DP size when using SP is (DP // SP). As SP sets of GPUs
have the same sequence but different parts

- Without multiproc

Train 1/2: 1%|▍ | 20/3934 [01:38<5:14:50, 4.83s/it

Avg GPU utilization: 55%

- With multiproc

Train 1/2: 1%|▍ | 20/3934 [01:00<2:57:09, 2.72s/it]

Avg GPU utilization: 95%

Co-authored-by: shengguangming <shengguangming@bytedance.com>

Close #119 

(not tested locally yet but this looks safe, especially given that there
are CI now)

- forbid uneven chunk for DataProto

- Set use_reentrant=False to avoid duplicate allgather in backward when
gradient checkpointing is enabled.
- Optimize temperature computation by using inplace op
- Fix testing logics

- Add format script
- Move save_checkpoint to a separate function
- Add timing/step, response_length/clip_ratio, prompt_length/clip_ratio
and critic/vf_explained_var metrics
- The training step starts from 1

- As titled