Files
foxhunt/crates/ml-ppo
jgrusewski 379c0bee37 fix(dqn): cast F32 constants to input dtype for BF16 mixed precision on H100
On H100 with BF16 mixed precision enabled, Candle's comparison ops (gt,
lt, le, ge) and arithmetic ops require matching dtypes. F32 threshold
constants compared against BF16 state tensors caused every training step
to fail with "dtype mismatch in cmp, lhs: BF16, rhs: F32".

Fixes:
- regime_conditional.rs: ADX/CUSUM thresholds cast to adx.dtype()
- quantile_regression.rs: Huber kappa, zero, and half tensors match
  input dtype (both quantile_huber_loss variants)
- rainbow_network.rs: LeakyReLU slope and ELU alpha/one tensors cast
  to x.dtype()
- continuous_ppo.rs: Huber delta/half tensors cast to abs_diff.dtype()

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-03-09 17:13:46 +01:00
..