tessl/pypi-chex
Comprehensive utilities library for JAX testing, debugging, and instrumentation
73
Evaluation results
100%
Model Output Validator
ULP-based floating-point comparisons (assert_trees_all_close_ulp)
Uses assert_trees_all_close_ulp
100%
100%
Passes max_ulp parameter
100%
100%
Handles tree structures
100%
100%
Error handling
100%
100%
Minimal implementation
100%
100%
100%
95%Hardware Compatibility Validator
Device and hardware assertions (assert_devices_available, assert_gpu_available, assert_tpu_available)
Uses assert_gpu_available
0%
100%
Uses assert_tpu_available
0%
100%
Uses assert_devices_available
0%
100%
Proper return values
100%
100%
100%
100%Array Shape Validator
Array and shape assertions (assert_shape, assert_rank, assert_equal_shape)
Uses assert_shape
0%
100%
Uses assert_rank
0%
100%
Uses assert_equal_shape
0%
100%
Wildcard pattern support
0%
100%
Proper error propagation
0%
100%
15%
15%Nested Model Comparison Validator
Path-based error reporting in nested structures
Uses tree comparison
0%
16%
Leverages path reporting
0%
0%
Handles structure types
0%
25%
Type checking
0%
33%
Error propagation
0%
0%
100%
Neural Network Parameter Validator
Value assertions for tree content (assert_tree_all_finite, assert_trees_all_equal)
assert_tree_all_finite usage
100%
100%
assert_trees_all_equal usage
100%
100%
Proper exception handling
100%
100%
Correct function signatures
100%
100%
0%
Shape Template Manager
Named dimension sizes (Dimensions class)
Uses Dimensions class
0%
0%
String-based retrieval
0%
0%
Flattening syntax
0%
0%
Wildcard support
0%
0%
56%
6%Multi-Device Parallel Testing Framework
Multi-device testing support (set_n_cpu_devices, CPU device simulation)
Device Setup Usage
100%
100%
Device Count Query
100%
100%
Parallel Mapping
0%
0%
Device Cleanup
0%
33%
Test Integration
50%
60%
55%
Neural Network Config Manager
JAX-friendly dataclasses (@dataclass, @mappable_dataclass)
Uses @dataclass decorator
100%
100%
Uses @mappable_dataclass decorator
0%
0%
Utilizes from_tuple method
50%
50%
JAX tree integration
100%
100%
Dictionary-style access
0%
0%
100%
78%Distributed Array Validator
Array sharding validation (assert_tree_is_sharded)
Uses assert_tree_is_sharded
0%
100%
Handles single arrays
20%
100%
Handles nested structures
25%
100%
Handles empty structures
100%
100%
Exception propagation
26%
100%
100%
50%Neural Network Layer Validator
Conditional assertions (if_args_not_none)
Uses if_args_not_none
0%
100%
Validates required parameters
100%
100%
Correct shape assertions
50%
100%
Validates dimension compatibility
100%
100%
Install with Tessl CLI
npx tessl i tessl/pypi-chex- Evaluated
- Agent
- Codex
Table of Contents