Create optimize utility/quick train and eval loop

[kmheckel]

Make spyx.optimize:

• [ ] surrogate gradient training loop
• [ ] take SNN, params, Dataset in form (#batches, batchsize, timesteps, ...), optimizer
• [ ] also take number of epochs, maybe eval frequency.