Skip to content

TIS support - #11

Open
Athe-kunal wants to merge 5 commits into
RiddleHe:masterfrom
Athe-kunal:master
Open

TIS support#11
Athe-kunal wants to merge 5 commits into
RiddleHe:masterfrom
Athe-kunal:master

Conversation

@Athe-kunal

Copy link
Copy Markdown

We are using vLLM for rollouts and FSDP; we need to fix it for TIS. This PR implements TIS

@Athe-kunal

Copy link
Copy Markdown
Author

@RiddleHe let me know about this PR

@Athe-kunal

Copy link
Copy Markdown
Author

Hi @RiddleHe

I added some more things to the PR

  1. Added MaxRL to the losses
  2. Eval with AIME 2025 (eval@k) (earlier it was empty)
  3. Added a script to generate the training for DAPO math
  4. Added wandb logging for training and evaluation

@Athe-kunal

Copy link
Copy Markdown
Author

@RiddleHe, any updates on this?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant