Skip to content

Move jax single gpu example - #447

Draft
hvdbm wants to merge 9 commits into
masterfrom
fix/move-jax-single-gpu-example
Draft

Move jax single gpu example#447
hvdbm wants to merge 9 commits into
masterfrom
fix/move-jax-single-gpu-example

Conversation

@hvdbm

@hvdbm hvdbm commented Jun 30, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Move the Jax Single GPU example under docs/examples/distributed/single_gpu/.
  • Add missing config .toml file in the Jax example.
  • Add a "Next Steps" sections to "Pytorch Setup" and "Jax Setup" pointing to the corresponding "Single-GPU Job".

Issues

@hvdbm hvdbm changed the title Fix/move jax single gpu example Move jax single gpu example Jun 30, 2026
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.

[DOC] - Move the second Jax example of Software Frameworks under Distributed Training/Single GPU Job

1 participant