Files
foxhunt/crates/ml/src
jgrusewski 4c57f3d704 fix(ml): DQN single-step action selection uses GPU argmax
Replace CPU-side argmax (to_vec1 + iter enumerate max_by) with
GPU-native argmax(D). Single u32 scalar extraction instead of
full Q-value vector transfer.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-02 18:03:20 +01:00
..