[TODO]: Clean hardcoded stuff
retraigo opened this issue · comments
Pranev (NeTT) commented
- Keep gradient descent as a separate function instead of hardcoding.
- Keep loss functions as separate functions instead of hardcoding.
- Allow both f32 and f64 inputs.