anton/fvnn-minijax
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Preliminary state

Browse Source
This commit is contained in:
Anton Pogrebnjak
2026-05-26 10:24:33 +02:00
commit 0ab11cfc5e
950 changed files with 6428 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/milestone1/open/fuzz/eval/activations/test.json
+15
View File
@@ -0,0 +1,15 @@
{
"command": "fuzz_eval",
"n_trials": 50,
"seed": 179320,
"primitives": [
"leaky_relu",
"elu",
"gelu",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted fuzz for new activation primitives (leaky_relu/elu/gelu) interleaved with add/mul; verifies eval on random leaky_relu slopes and elementwise-preserving shapes",
"points": 6
}
tests/milestone1/open/fuzz/eval/all_primitives/test.json
+31
View File
@@ -0,0 +1,31 @@
{
"command": "fuzz_eval",
"n_trials": 100,
"seed": 5024369,
"primitives": [
"neg",
"reciprocal",
"relu",
"square",
"sqrt",
"exp",
"log",
"leaky_relu",
"elu",
"gelu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "fuzz forward-pass with all supported primitives to verify robustness including numerically unsafe ops",
"points": 6
}
tests/milestone1/open/fuzz/eval/axis_ops/test.json
+15
View File
@@ -0,0 +1,15 @@
{
"command": "fuzz_eval",
"n_trials": 50,
"seed": 2867273,
"primitives": [
"pad",
"conv",
"avgpool",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted fuzz for last-axis ops (pad with dilation, conv, avgpool) to stress length arithmetic and kernel/window/stride combinations",
"points": 6
}
tests/milestone1/open/fuzz/eval/broad/test.json
+31
View File
@@ -0,0 +1,31 @@
{
"command": "fuzz_eval",
"n_trials": 200,
"seed": 8690109,
"primitives": [
"neg",
"reciprocal",
"relu",
"square",
"sqrt",
"exp",
"log",
"leaky_relu",
"elu",
"gelu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "broad forward-pass fuzz: 200 trials over all primitives to stress rare shape/dtype combinations",
"points": 6
}
tests/milestone1/open/fuzz/eval/linalg/test.json
+13
View File
@@ -0,0 +1,13 @@
{
"command": "fuzz_eval",
"n_trials": 50,
"seed": 9417485,
"primitives": [
"dot",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted fuzz for dot/add/mul to stress broadcasting and matmul shape logic",
"points": 6
}
tests/milestone1/open/fuzz/eval/numerics/test.json
+25
View File
@@ -0,0 +1,25 @@
{
"command": "fuzz_eval",
"n_trials": 100,
"seed": 1677293,
"primitives": [
"neg",
"relu",
"square",
"leaky_relu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": true,
"description": "forward-pass numerics fuzz: safe primitives only, fails on NaN/Inf in outputs",
"points": 6
}
tests/milestone1/open/fuzz/eval/safe_primitives/test.json
+25
View File
@@ -0,0 +1,25 @@
{
"command": "fuzz_eval",
"n_trials": 150,
"seed": 7119429,
"primitives": [
"neg",
"relu",
"square",
"leaky_relu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "fuzz forward-pass with safe primitives (excludes log/sqrt/reciprocal/exp) to verify no crashes or shape errors",
"points": 6
}
tests/milestone1/open/fuzz/eval/shape_ops/test.json
+14
View File
@@ -0,0 +1,14 @@
{
"command": "fuzz_eval",
"n_trials": 50,
"seed": 5257192,
"primitives": [
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum"
],
"check_nan_inf": false,
"description": "targeted fuzz for shape-manipulating primitives (expand_dims/moveaxis/reshape/reduce_sum) to catch shape-propagation bugs",
"points": 6
}
tests/milestone1/open/fuzz/eval/unary_unsafe/test.json
+16
View File
@@ -0,0 +1,16 @@
{
"command": "fuzz_eval",
"n_trials": 50,
"seed": 2294743,
"primitives": [
"log",
"sqrt",
"reciprocal",
"exp",
"neg",
"mul"
],
"check_nan_inf": false,
"description": "targeted crash-only fuzz for numerically unsafe unaries (log/sqrt/reciprocal/exp) \u2014 NaN/Inf outputs permitted, only shape/crash failures detected",
"points": 6
}
tests/milestone1/open/fuzz/grad/activations/test.json
+15
View File
@@ -0,0 +1,15 @@
{
"command": "fuzz_grad",
"n_trials": 50,
"seed": 9941134,
"primitives": [
"leaky_relu",
"elu",
"gelu",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted gradient fuzz for new activation primitives; exercises backward rules for leaky_relu (piecewise), elu (exp branch), gelu (erf+gaussian)",
"points": 6
}
tests/milestone1/open/fuzz/grad/all_primitives/test.json
+31
View File
@@ -0,0 +1,31 @@
{
"command": "fuzz_grad",
"n_trials": 100,
"seed": 9326167,
"primitives": [
"neg",
"reciprocal",
"relu",
"square",
"sqrt",
"exp",
"log",
"leaky_relu",
"elu",
"gelu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "fuzz gradient computation with all supported primitives to verify backward-pass robustness",
"points": 6
}
tests/milestone1/open/fuzz/grad/axis_ops/test.json
+15
View File
@@ -0,0 +1,15 @@
{
"command": "fuzz_grad",
"n_trials": 50,
"seed": 1547885,
"primitives": [
"pad",
"conv",
"avgpool",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted gradient fuzz for last-axis ops (pad/conv/avgpool); exercises strided/dilated backward accumulation",
"points": 6
}
tests/milestone1/open/fuzz/grad/broad/test.json
+31
View File
@@ -0,0 +1,31 @@
{
"command": "fuzz_grad",
"n_trials": 200,
"seed": 6003614,
"primitives": [
"neg",
"reciprocal",
"relu",
"square",
"sqrt",
"exp",
"log",
"leaky_relu",
"elu",
"gelu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "broad backward-pass fuzz: 1000 trials over all primitives to stress gradient shape-propagation",
"points": 6
}
tests/milestone1/open/fuzz/grad/linalg/test.json
+13
View File
@@ -0,0 +1,13 @@
{
"command": "fuzz_grad",
"n_trials": 50,
"seed": 4046095,
"primitives": [
"dot",
"add",
"mul"
],
"check_nan_inf": false,
"description": "targeted gradient fuzz for dot/add/mul; exercises broadcasting backward rules",
"points": 6
}
tests/milestone1/open/fuzz/grad/numerics/test.json
+16
View File
@@ -0,0 +1,16 @@
{
"command": "fuzz_grad",
"n_trials": 100,
"seed": 4302181,
"primitives": [
"relu",
"square",
"dot",
"add",
"mul",
"neg"
],
"check_nan_inf": true,
"description": "backward-pass numerics fuzz over curated finite-gradient subset (relu/square/dot/add/mul/neg); fails on NaN/Inf in gradients",
"points": 6
}
tests/milestone1/open/fuzz/grad/safe_primitives/test.json
+25
View File
@@ -0,0 +1,25 @@
{
"command": "fuzz_grad",
"n_trials": 150,
"seed": 806830,
"primitives": [
"neg",
"relu",
"square",
"leaky_relu",
"add",
"mul",
"dot",
"where",
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum",
"pad",
"conv",
"avgpool"
],
"check_nan_inf": false,
"description": "fuzz gradient computation with safe primitives to verify no crashes or shape errors",
"points": 6
}
tests/milestone1/open/fuzz/grad/shape_ops/test.json
+14
View File
@@ -0,0 +1,14 @@
{
"command": "fuzz_grad",
"n_trials": 50,
"seed": 5785067,
"primitives": [
"expand_dims",
"moveaxis",
"reshape",
"reduce_sum"
],
"check_nan_inf": false,
"description": "targeted gradient fuzz for shape-manipulating primitives to verify transposed shape-propagation in the backward pass",
"points": 6
}
tests/milestone1/open/fuzz/grad/unary_unsafe/test.json
+16
View File
@@ -0,0 +1,16 @@
{
"command": "fuzz_grad",
"n_trials": 50,
"seed": 9102614,
"primitives": [
"log",
"sqrt",
"reciprocal",
"exp",
"neg",
"mul"
],
"check_nan_inf": false,
"description": "targeted crash-only gradient fuzz for unsafe unaries; NaN/Inf gradients permitted",
"points": 6
}