Estimate Critical Batch Size | TensorTonic