Skip to content

Add routed_expert_logits support to R3 datums#5

Closed
kiddyboots216 wants to merge 2 commits into
mainfrom
issue-4-routed-expert-logits
Closed

Add routed_expert_logits support to R3 datums#5
kiddyboots216 wants to merge 2 commits into
mainfrom
issue-4-routed-expert-logits

Conversation

@kiddyboots216
Copy link
Copy Markdown
Contributor

@kiddyboots216 kiddyboots216 commented Apr 10, 2026

Summary

  • add optional routed_expert_logits support to xorl_client.types.Datum
  • preserve and lift both R3 routing fields during TrainingClient datum conversion so request payloads include routed_experts and routed_expert_logits
  • update chunking tests to follow MAX_CHUNK_LEN so they exercise the current multi-chunk behavior consistently
  • add tests covering datum serialization and forward_backward request construction for both routing fields

Closes #4.

Testing

  • python -m pytest -q /home/apanda/xorl-client-issue-4/tests/test_types.py /home/apanda/xorl-client-issue-4/tests/test_chunked_forward_backward.py

@kiddyboots216
Copy link
Copy Markdown
Contributor Author

Folded into #2. The examples/filler-rl-pipeline branch now contains the routed_expert_logits changes plus the GSM8K/pipeline and worker_port fixes, so I am closing this standalone PR as superseded.

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.

Add routed_expert_logits support to xorl-client R3 datums

1 participant