2.18.1

Various fixes & improvements

• ref(batching): add compute_batch_size to BatchStep (#390) by @MeredithAnya
• ref(rust/run_task): Remove unnecessary boxing, make FnMut, change return type (#388) by @untitaker