Category,File,Line,Function,Current Code,Fixed Code,Priority,Estimated Time (min)
Model Init,ml/src/mamba/mod.rs,394,Mamba2SSM::new,"let device = Device::Cpu;","Add device: &Device parameter",CRITICAL,30
State Init,ml/src/mamba/mod.rs,222,Mamba2State::zeros,"let device = match Device::cuda_if_available(0)","Accept device: &Device parameter",HIGH,20
SSD Layer Init,ml/src/mamba/ssd_layer.rs,62,SSDLayer::new,"let device = Device::Cpu;","Add device: &Device parameter",CRITICAL,30
SSD Caller,ml/src/mamba/mod.rs,416,Mamba2SSM::new,"SSDLayer::new(&config, i)?","SSDLayer::new(&config, i, &device)?",HIGH,5
State Caller,ml/src/mamba/mod.rs,446,Mamba2SSM::new,"Mamba2State::zeros(&config)?","Mamba2State::zeros(&config, &device)?",HIGH,5
LR Schedule,ml/src/mamba/mod.rs,1156,update_learning_rate,"Tensor::new(&[step], &Device::Cpu)?","Tensor::new(&[step], self.device())?",HIGH,10
Grad Clip,ml/src/mamba/mod.rs,1417,clip_gradients,"Tensor::new(&[clip_factor], &Device::Cpu)?","Tensor::new(&[clip_factor], self.device())?",HIGH,5
Adam Beta1,ml/src/mamba/mod.rs,1508,optimizer_step,"Tensor::new(&[beta1], &Device::Cpu)?","Tensor::new(&[beta1], device)?",HIGH,5
Adam 1-Beta1,ml/src/mamba/mod.rs,1509,optimizer_step,"Tensor::new(&[1.0-beta1], &Device::Cpu)?","Tensor::new(&[1.0-beta1], device)?",HIGH,5
Adam Beta2,ml/src/mamba/mod.rs,1515,optimizer_step,"Tensor::new(&[beta2], &Device::Cpu)?","Tensor::new(&[beta2], device)?",HIGH,5
Adam 1-Beta2,ml/src/mamba/mod.rs,1516,optimizer_step,"Tensor::new(&[1.0-beta2], &Device::Cpu)?","Tensor::new(&[1.0-beta2], device)?",HIGH,5
Bias Corr1,ml/src/mamba/mod.rs,1523,optimizer_step,"Tensor::new(&[bias_correction1], &Device::Cpu)?","Tensor::new(&[bias_correction1], device)?",HIGH,5
Bias Corr2,ml/src/mamba/mod.rs,1524,optimizer_step,"Tensor::new(&[bias_correction2], &Device::Cpu)?","Tensor::new(&[bias_correction2], device)?",HIGH,5
Adam Epsilon,ml/src/mamba/mod.rs,1529,optimizer_step,"Tensor::new(&[eps], &Device::Cpu)?","Tensor::new(&[eps], device)?",HIGH,5
Adam LR,ml/src/mamba/mod.rs,1530,optimizer_step,"Tensor::new(&[lr], &Device::Cpu)?","Tensor::new(&[lr], device)?",HIGH,5
Delta Min,ml/src/mamba/mod.rs,1563,optimizer_step,"Tensor::new(&[1e-6], &Device::Cpu)?","Tensor::new(&[1e-6], device)?",HIGH,5
Delta Max,ml/src/mamba/mod.rs,1564,optimizer_step,"Tensor::new(&[1.0], &Device::Cpu)?","Tensor::new(&[1.0], device)?",HIGH,5
Weight Decay,ml/src/mamba/mod.rs,1500,optimizer_step,"Tensor::new(&[wd], &Device::Cpu)?","Tensor::new(&[wd], device)?",HIGH,5
Inference,ml/src/mamba/mod.rs,670,predict_single_fast,"let device = &Device::Cpu;","let device = self.device();",MEDIUM,10
Device Helper,ml/src/mamba/mod.rs,N/A,Mamba2SSM,"N/A","fn device(&self) -> &Device { self.input_projection.ws().device() }",HIGH,15
Trainer,ml/src/trainers/mamba2.rs,299,Mamba2Trainer::new,"Mamba2SSM::new(config)?","Mamba2SSM::new(config, &device)?",HIGH,5
Default HFT,ml/src/mamba/mod.rs,494,Mamba2SSM::default_hft,"Self::new(config)","Self::new(config, device)",MEDIUM,10
Selective State,ml/src/mamba/selective_state.rs,TBD,SelectiveStateSpace::new,TBD,TBD,MEDIUM,60