Skip to content

Use 'tensors on device' wording in JAX starters#279

Merged
kunal-mansukhani merged 1 commit into
mainfrom
jax-starter-device-wording
Jun 8, 2026
Merged

Use 'tensors on device' wording in JAX starters#279
kunal-mansukhani merged 1 commit into
mainfrom
jax-starter-device-wording

Conversation

@kunal-mansukhani

Copy link
Copy Markdown
Contributor

Summary

  • Normalize all JAX starter parameter comments from GPU-specific phrasing to the device-neutral "on device", since the JAX/XLA reference path targets TPU as well as GPU.
  • Sweeps all four phrasings that existed across the 86 JAX starters: are tensors on GPU, are tensors on the GPU, is a tensor on GPU, is a tensor on the GPU... on device.
  • Update CLAUDE.md's comment-template table and starter checklist so new JAX starters follow this convention.

Scope

  • Only .jax.py starters changed. PyTorch/Triton/CuTe ("tensors on the GPU") and CUDA/Mojo ("device pointers") are intentionally left as-is.
  • No behavioral change — comments only.

🤖 Generated with Claude Code

Normalize all JAX starter parameter comments from GPU-specific
phrasing ("tensors on the GPU" / "tensor on GPU") to the
device-neutral "on device", matching the XLA/TPU runner. Update
CLAUDE.md's comment template and checklist accordingly.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@kunal-mansukhani kunal-mansukhani merged commit b17798f into main Jun 8, 2026
4 checks passed
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