CtrlK
BlogDocsLog inGet started
Tessl Logo

tessl/pypi-chex

Comprehensive utilities library for JAX testing, debugging, and instrumentation

73

1.92x
Overview
Eval results
Files

rubric.jsonevals/scenario-2/

{
  "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-chex

tile.json