Comprehensive utilities library for JAX testing, debugging, and instrumentation
73
{
"context": "This criteria evaluates how well the engineer uses chex's device and hardware assertion functions to validate GPU, TPU, and general device availability. The focus is on proper usage of chex.assert_gpu_available(), chex.assert_tpu_available(), and chex.assert_devices_available() functions.",
"type": "weighted_checklist",
"checklist": [
{
"name": "Uses assert_gpu_available",
"description": "The check_gpu_available() function correctly calls chex.assert_gpu_available() to validate GPU availability",
"max_score": 35
},
{
"name": "Uses assert_tpu_available",
"description": "The check_tpu_available() function correctly calls chex.assert_tpu_available() to validate TPU availability",
"max_score": 35
},
{
"name": "Uses assert_devices_available",
"description": "The validate_devices() function correctly calls chex.assert_devices_available() with the appropriate device specifications to validate multiple device requirements",
"max_score": 25
},
{
"name": "Proper return values",
"description": "Functions return True when assertions pass successfully, following the specified API contract",
"max_score": 5
}
]
}Install with Tessl CLI
npx tessl i tessl/pypi-chexevals
scenario-1
scenario-2
scenario-3
scenario-4
scenario-5
scenario-6
scenario-7
scenario-8
scenario-9
scenario-10