perf: add class balancing and label smoothing to the loss function per batch

perf: tweak hyper-params

perf: replace metal with wgpu support and add http file for predict request

feat: scaffold model, training, inference and api