计算 IoU (Intersection over Union) boxA, boxB 格式: [x1, y1, x2, y2]

correct example into "messages" eg: { "system": "You are a helpful assistant.", "conversations": [ {"from": "user"

Ensure the CSV file exists by converting from Parquet if necessary.

Requests to the IOI package return the score as a float in the stdout, as well as optional feedback/errors in stderr. Returns a tuple

Extract answer from query results, specifically looking for: - Numbers within asterisks - Yes/No answers in various formats Args:

Extract image bytes from string using regex.

vLLM enforces a constraint that the number of attention heads must be divisible by the number of GPUs and 64 must be divisible by the number of GPUs.

vLLM enforces a constraint that the number of attention heads must be divisible by the number of GPUs and 64 must be divisible by the number of GPUs.

Function to get model param counts from safetensors metadata or find patterns like 42m, 1.5b, 0.5m or products like 8x7b in a repo ID.

Function to get model param counts from safetensors metadata or find patterns like 42m, 1.5b, 0.5m or products like 8x7b in a repo ID.

Get list of active piston worker endpoints from squeue output

init a model(args.model_path) on a specific gpu

Reward function that evaluates IOI problems using Piston+our IOI package. Assumes the dataset has the same format as hf.co/datasets/open-r1/ioi

Load IOI tests for a given year and problem id.

Load IOI tests for a given year.

let each gpu (along with a model) answer a chunk of questions

Process dataset and generate conversation JSON files.