## Summary

Brief description of what this PR does and why.

Closes #ISSUE_NUMBER

## Changes

- ...
- ...

## Testing

- [ ] `cargo build` passes
- [ ] `cargo clippy --workspace` passes
- [ ] Tested on: [OS, GPU, CUDA version]

## Notes

Any additional context for reviewers.
