{ "version": "0.32.2", "device": "Device(cpu, 0)", "cpu_seconds": 0.666912, "forward_scenarios": 192, "gradient_scenarios": 12, "summary": { "before": { "checks": 2760, "failed": 102 }, "after": { "checks": 2760, "failed": 0 } }, "checks": [ { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0166015625, -1.0546875, -0.254150390625, -0.263671875, 1.271484375, 1.3193359375 ], "expected": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "expected": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.267333984375, -0.267333984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0166015625, -1.0546875, -0.254150390625, -0.263671875, 1.271484375, 1.3193359375 ], "expected": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003254413604736328, 0.0006508827209472656 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "expected": [ -1.0166015625, -1.0556640625, -0.254150390625, -0.263916015625, 1.271484375, 1.3193359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26053866744041443, 0.5210773348808289 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838596105575562, 0.8385633230209351 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.988491177558899, -3.8450770378112793, -0.8590123653411865, -1.6610426902770996, 1.3999452590942383, 2.7070260047912598 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.399752140045166, 2.7066047191619873 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.27490234375, 0.916015625, -0.1312255859375, 0.32275390625, 2.1572265625, -0.864501953125 ], "expected": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "expected": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.1510009765625, 0.32550048828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.27490234375, 0.916015625, -0.1312255859375, 0.32275390625, 2.1572265625, -0.864501953125 ], "expected": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003254413604736328, 0.0006508827209472656 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "expected": [ -1.27490234375, 0.916748046875, -0.1312255859375, 0.32293701171875, 2.1572265625, -0.864501953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26053866744041443, 0.5210773348808289 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838596105575562, 0.8385633230209351 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.732736825942993, 3.00880765914917, -1.0385185480117798, 1.3707820177078247, 2.3499178886413574, -1.9052696228027344 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7329931259155273, 3.009004592895508, -1.0387859344482422, 1.3710185289382935, 2.349628210067749, -1.9049534797668457 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.26708984375, -0.26708984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.267333984375, -0.267333984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.26708984375, -0.26708984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041656494140625, 0.08331298828125 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.166748046875, 2.0419921875 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2967033386230469, 0.5934066772460938 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.03912353515625, 1.859619140625 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7625468969345093, -2.6350765228271484, -0.7815575003623962, -1.1684590578079224, 1.180421233177185, 1.7647758722305298 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7627670764923096, -2.635500907897949, -0.78174889087677, -1.1687873601913452, 1.180287480354309, 1.7646397352218628 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.150634765625, 0.3253173828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.1510009765625, 0.32550048828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.150634765625, 0.3253173828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041656494140625, 0.08331298828125 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.166748046875, 2.0419921875 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2967033386230469, 0.5934066772460938 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.03912353515625, 1.859619140625 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.393820285797119, 2.1013073921203613, -0.922336220741272, 1.0013443231582642, 2.020631790161133, -1.1985819339752197 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.394150733947754, 2.101625680923462, -0.9226233959197998, 1.0015904903411865, 2.0204312801361084, -1.1984797716140747 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.267333984375, -0.267333984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.6640625, 21.328125 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.59130859375, 19.1826171875 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, -0.0, -0.0 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804812490940094, -0.06258240342140198, -0.06258348375558853 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.1510009765625, 0.32550048828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.6640625, 21.328125 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.59130859375, 19.1826171875 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.12133154273033142, 0.18933533132076263, 0.13292983174324036, 0.183536097407341, 0.15612639486789703, 0.17193761467933655 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, -1.068359375, -0.267333984375, -0.267333984375, 1.3369140625, 1.3369140625 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.25, 682.5 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.064453125, -1.064453125, -0.26318359375, -0.26318359375, 1.3408203125, 1.3408203125 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.85400390625, 597.7080078125 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "expected": [ -1.0693359375, -1.0693359375, -0.267333984375, -0.267333984375, 1.3359375, 1.3359375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, -0.0, -0.0 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3525390625, 0.92626953125, -0.1510009765625, 0.32550048828125, 2.25537109375, -0.877685546875 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.25, 682.5 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.3466796875, 0.92333984375, -0.144775390625, 0.3223876953125, 2.26123046875, -0.880615234375 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.85400390625, 597.7080078125 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "expected": [ -1.35400390625, 0.927001953125, -0.1510009765625, 0.32550048828125, 2.25390625, -0.876953125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09953396022319794, -0.09953510761260986, -0.09322060644626617, -0.09322167932987213, -0.08059389889240265, -0.08059483021497726, -0.1058473140001297, -0.105848528444767, -0.08690725266933441, -0.0869082510471344, -0.07428054511547089, -0.07428140193223953 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.10069906711578369, 0.1996513307094574, 0.11016908288002014, 0.1949162632226944, 0.12910914421081543, 0.1854461133480072, 0.09122902154922485, 0.2043863981962204, 0.11963912844657898, 0.1901811957359314, 0.13857918977737427, 0.1807110607624054 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893399685621262, -0.08893399685621262, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11689496040344238, 0.1915525197982788, 0.1171908974647522, 0.1914045512676239, 0.11778278648853302, 0.1911086142063141, 0.11659900844097137, 0.19170048832893372, 0.11748683452606201, 0.191256582736969, 0.11807872354984283, 0.19096064567565918 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "expected": [ -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625, -0.8544921875, -0.8720703125, -0.284912109375, -0.290771484375, 0.8544921875, 0.8720703125, -1.4248046875, -1.4541015625, 0.284912109375, 0.290771484375, 1.4248046875, 1.4541015625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905027866363525, 0.8651362061500549 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9863195419311523, -3.797421455383301, -0.8615965247154236, -1.647189736366272, 1.3878493309020996, 2.653273344039917, -3.1110424995422363, -5.947652816772461, 0.263126403093338, 0.5030418634414673, 2.5125722885131836, 4.803504943847656, -1.9863195419311523, -3.797421455383301, -0.8615965247154236, -1.647189736366272, 1.3878493309020996, 2.653273344039917, -3.1110424995422363, -5.947652816772461, 0.263126403093338, 0.5030418634414673, 2.5125722885131836, 4.803504943847656 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "expected": [ -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875, -1.03173828125, 0.779052734375, -0.1773681640625, 0.34307861328125, 1.53173828125, -0.529052734375, -1.88720703125, 1.215576171875, 0.6773681640625, -0.09307861328125, 2.38720703125, -0.965576171875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905027866363525, 0.8651362061500549 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7294793128967285, 2.9730660915374756, -1.042394757270813, 1.3603923320770264, 2.3317739963531494, -1.864954948425293, -4.416563987731934, 4.585739612579346, 0.6446896195411682, -0.25228139758110046, 4.018858432769775, -3.477628707885742, -2.7294793128967285, 2.9730660915374756, -1.042394757270813, 1.3603923320770264, 2.3317739963531494, -1.864954948425293, -4.416563987731934, 4.585739612579346, 0.6446896195411682, -0.25228139758110046, 4.018858432769775, -3.477628707885742 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.272705078125, 2.4658203125 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384735107421875, 2.25701904296875 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7060140371322632, -2.4233012199401855, -0.7688046097755432, -1.092045545578003, 1.1056143045425415, 1.5704655647277832, -2.643223524093628, -3.754556655883789, 0.16840481758117676, 0.23920997977256775, 2.0428237915039062, 2.9017210006713867, -1.7060140371322632, -2.4233012199401855, -0.7688046097755432, -1.092045545578003, 1.1056143045425415, 1.5704655647277832, -2.643223524093628, -3.754556655883789, 0.16840481758117676, 0.23920997977256775, 2.0428237915039062, 2.9017210006713867 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668, -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.272705078125, 2.4658203125 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384735107421875, 2.25701904296875 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.30902099609375, 1.9424759149551392, -0.9032069444656372, 0.9440341591835022, 1.908421516418457, -1.0528491735458374, -3.7148351669311523, 2.940917491912842, 0.5026072263717651, -0.054407477378845215, 3.3142356872558594, -2.05129075050354, -2.30902099609375, 1.9424759149551392, -0.9032069444656372, 0.9440341591835022, 1.908421516418457, -1.0528491735458374, -3.7148351669311523, 2.940917491912842, 0.5026072263717651, -0.054407477378845215, 3.3142356872558594, -2.05129075050354 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251, -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0, -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0, -0.0, -0.0, -0.0, -0.0, 0.0, 0.0, -0.0, -0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162, -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125, 0.25, 0.125 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872, 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, "NaN", -0.0, "NaN", 0.0, "NaN", -0.0, "NaN", 0.0, "NaN", 0.0, "NaN", -0.0, "NaN", -0.0, "NaN", 0.0, "NaN", -0.0, "NaN", 0.0, "NaN", 0.0, "NaN" ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ -0.0, "NaN", -0.0, "NaN", 0.0, "NaN", -0.0, "NaN", 0.0, "NaN", 0.0, "NaN", -0.0, "NaN", -0.0, "NaN", 0.0, "NaN", -0.0, "NaN", 0.0, "NaN", 0.0, "NaN" ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": false, "actual": [ 512.0, "Infinity" ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.8779296875, -0.8779296875, -0.292724609375, -0.292724609375, 0.8779296875, 0.8779296875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": false, "actual": [ 448.265625, "Infinity" ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "expected": [ -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875, -0.87841796875, -0.87841796875, -0.292724609375, -0.292724609375, 0.87841796875, 0.87841796875, -1.4638671875, -1.4638671875, 0.292724609375, 0.292724609375, 1.4638671875, 1.4638671875 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN", -0.0, "NaN" ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN" ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": false, "actual": [ 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN" ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": false, "actual": [ 512.0, "Infinity" ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.06689453125, 0.783447265625, -0.1890869140625, 0.34454345703125, 1.56689453125, -0.533447265625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": false, "actual": [ 448.265625, "Infinity" ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "expected": [ -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625, -1.067626953125, 0.7838134765625, -0.1890869140625, 0.34454345703125, 1.567626953125, -0.5338134765625, -1.94580078125, 1.222900390625, 0.6890869140625, -0.09454345703125, 2.44580078125, -0.972900390625 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": false, "actual": [ 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN", 0.25, "NaN" ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474, 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474 ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 0.004, "atol": 4e-06 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.265625, 1.3125 ], "expected": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "expected": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.265625, -0.265625, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.265625, 1.3125 ], "expected": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003261566162109375, 0.000652313232421875 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750177621841431, 0.8750710487365723 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "expected": [ -1.015625, -1.0546875, -0.25390625, -0.263671875, 1.2734375, 1.3203125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26200413703918457, 0.5240082740783691 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7839511036872864, 0.8389293551445007 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9900301694869995, -3.847437858581543, -0.8606172800064087, -1.6638799905776978, 1.398208498954773, 2.703235387802124 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.399752140045166, 2.7066047191619873 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.1484375, -0.859375 ], "expected": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "expected": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1484375, 0.32421875, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.1484375, -0.859375 ], "expected": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003261566162109375, 0.000652313232421875 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750177621841431, 0.8750710487365723 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "expected": [ -1.2734375, 0.916015625, -0.130859375, 0.32275390625, 2.16015625, -0.865234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26200413703918457, 0.5240082740783691 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7839511036872864, 0.8389293551445007 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7350451946258545, 3.0105783939361572, -1.0409259796142578, 1.3729100227355957, 2.3473126888275146, -1.9024264812469482 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7329931259155273, 3.009004592895508, -1.0387859344482422, 1.3710185289382935, 2.349628210067749, -1.9049534797668457 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.265625, -0.265625, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041748046875, 0.08349609375 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.166015625, 2.0390625 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.298248291015625, 0.59649658203125 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.03857421875, 1.857421875 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7645291090011597, -2.638901710510254, -0.7832801938056946, -1.1714171171188354, 1.1792174577713013, 1.7635520696640015 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7627670764923096, -2.635500907897949, -0.78174889087677, -1.1687873601913452, 1.180287480354309, 1.7646397352218628 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1484375, 0.32421875, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041748046875, 0.08349609375 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.166015625, 2.0390625 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.298248291015625, 0.59649658203125 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.03857421875, 1.857421875 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.3967936038970947, 2.1041762828826904, -0.9249203205108643, 1.0035628080368042, 2.0188262462615967, -1.1976640224456787 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.394150733947754, 2.101625680923462, -0.9226233959197998, 1.0015904903411865, 2.0204312801361084, -1.1984797716140747 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.265625, -0.265625, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.6875, 21.375 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19072.875, 76288.875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.61328125, 19.2265625 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16688.783203125, 66752.8359375 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.08602605015039444, -0.08602753281593323, -0.07828521728515625, -0.0782865658402443, -0.06280355155467987, -0.06280463188886642 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804812490940094, -0.06258240342140198, -0.06258348375558853 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1484375, 0.32421875, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.6875, 21.375 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19072.875, 76288.875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.61328125, 19.2265625 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16688.783203125, 66752.8359375 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.12096092104911804, 0.18952065706253052, 0.13257217407226562, 0.18371492624282837, 0.1557946801185608, 0.17210346460342407 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.12133154273033142, 0.18933533132076263, 0.13292983174324036, 0.183536097407341, 0.15612639486789703, 0.17193761467933655 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.265625, -0.265625, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 342.0, 684.0 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19529728.0, 78118912.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.1015625, -1.1015625, -0.30078125, -0.30078125, 1.3046875, 1.3046875 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 299.51171875, 599.0234375 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17088512.0, 68354048.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "expected": [ -1.0703125, -1.0703125, -0.267578125, -0.267578125, 1.3359375, 1.3359375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.07281675189733505, -0.07281675189733505, -0.07257484644651413, -0.07257484644651413, -0.07209103554487228, -0.07209103554487228 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1484375, 0.32421875, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 342.0, 684.0 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19529728.0, 78118912.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": false, "actual": [ -1.40234375, 0.951171875, -0.201171875, 0.3505859375, 2.20703125, -0.853515625 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 299.51171875, 599.0234375 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17088512.0, 68354048.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "expected": [ -1.35546875, 0.927734375, -0.1513671875, 0.32568359375, 2.25390625, -0.876953125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.14077487587928772, 0.17961256206035614, 0.1411377340555191, 0.17943114042282104, 0.14186345040798187, 0.17906826734542847 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09953396022319794, -0.09953510761260986, -0.09322060644626617, -0.09322167932987213, -0.08059389889240265, -0.08059483021497726, -0.1058473140001297, -0.105848528444767, -0.08690725266933441, -0.0869082510471344, -0.07428054511547089, -0.07428140193223953 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09953396022319794, -0.09953510761260986, -0.09322060644626617, -0.09322167932987213, -0.08059389889240265, -0.08059483021497726, -0.1058473140001297, -0.105848528444767, -0.08690725266933441, -0.0869082510471344, -0.07428054511547089, -0.07428140193223953 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.10069906711578369, 0.1996513307094574, 0.11016908288002014, 0.1949162632226944, 0.12910914421081543, 0.1854461133480072, 0.09122902154922485, 0.2043863981962204, 0.11963912844657898, 0.1901811957359314, 0.13857918977737427, 0.1807110607624054 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.10069906711578369, 0.1996513307094574, 0.11016908288002014, 0.1949162632226944, 0.12910914421081543, 0.1854461133480072, 0.09122902154922485, 0.2043863981962204, 0.11963912844657898, 0.1901811957359314, 0.13857918977737427, 0.1807110607624054 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893399685621262, -0.08893399685621262, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893399685621262, -0.08893399685621262, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.11689496040344238, 0.1915525197982788, 0.1171908974647522, 0.1914045512676239, 0.11778278648853302, 0.1911086142063141, 0.11659900844097137, 0.19170048832893372, 0.11748683452606201, 0.191256582736969, 0.11807872354984283, 0.19096064567565918 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11689496040344238, 0.1915525197982788, 0.1171908974647522, 0.1914045512676239, 0.11778278648853302, 0.1911086142063141, 0.11659900844097137, 0.19170048832893372, 0.11748683452606201, 0.191256582736969, 0.11807872354984283, 0.19096064567565918 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125, -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125, -0.85546875, -0.8671875, -0.28515625, -0.2890625, 0.85546875, 0.8671875, -1.421875, -1.4453125, 0.28515625, 0.2890625, 1.421875, 1.4453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750243186950684, 0.8750972747802734 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "expected": [ -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125, -0.85546875, -0.87109375, -0.28515625, -0.291015625, 0.85546875, 0.87109375, -1.421875, -1.453125, 0.28515625, 0.291015625, 1.421875, 1.453125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905486226081848, 0.8653194904327393 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9862618446350098, -3.7970192432403564, -0.8615715503692627, -1.6470153331756592, 1.3878090381622314, 2.6529924869537354, -3.110952138900757, -5.947022914886475, 0.26311877369880676, 0.5029885768890381, 2.5124993324279785, 4.802996635437012, -1.9862618446350098, -3.7970192432403564, -0.8615715503692627, -1.6470153331756592, 1.3878090381622314, 2.6529924869537354, -3.110952138900757, -5.947022914886475, 0.26311877369880676, 0.5029885768890381, 2.5124993324279785, 4.802996635437012 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375, -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375, -1.033203125, 0.775390625, -0.177734375, 0.341796875, 1.533203125, -0.525390625, -1.8828125, 1.208984375, 0.677734375, -0.091796875, 2.3828125, -0.958984375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750243186950684, 0.8750972747802734 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "expected": [ -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375, -1.033203125, 0.7783203125, -0.177734375, 0.34326171875, 1.533203125, -0.5283203125, -1.8828125, 1.21484375, 0.677734375, -0.09326171875, 2.3828125, -0.96484375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905486226081848, 0.8653194904327393 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7293927669525146, 2.972764492034912, -1.042357325553894, 1.3602614402770996, 2.3317136764526367, -1.8647444248199463, -4.416428089141846, 4.585267066955566, 0.6446781754493713, -0.25224143266677856, 4.018749237060547, -3.477247476577759, -2.7293927669525146, 2.972764492034912, -1.042357325553894, 1.3602614402770996, 2.3317136764526367, -1.8647444248199463, -4.416428089141846, 4.585267066955566, 0.6446781754493713, -0.25224143266677856, 4.018749237060547, -3.477247476577759 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2734375, 2.46875 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.13916015625, 2.259765625 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7054997682571411, -2.421828031539917, -0.7685728669166565, -1.0913816690444946, 1.1052809953689575, 1.569510817527771, -2.6424267292022705, -3.75227427482605, 0.16835404932498932, 0.239064559340477, 2.042207956314087, 2.8999569416046143, -1.7054997682571411, -2.421828031539917, -0.7685728669166565, -1.0913816690444946, 1.1052809953689575, 1.569510817527771, -2.6424267292022705, -3.75227427482605, 0.16835404932498932, 0.239064559340477, 2.042207956314087, 2.8999569416046143 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668, -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2734375, 2.46875 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.13916015625, 2.259765625 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.3082497119903564, 1.941370964050293, -0.9028593301773071, 0.9435362815856934, 1.907921552658081, -1.0521330833435059, -3.7136402130126953, 2.9392056465148926, 0.5025310516357422, -0.054298415780067444, 3.31331205368042, -2.0499677658081055, -2.3082497119903564, 1.941370964050293, -0.9028593301773071, 0.9435362815856934, 1.907921552658081, -1.0521330833435059, -3.7136402130126953, 2.9392056465148926, 0.5025310516357422, -0.054298415780067444, 3.31331205368042, -2.0499677658081055 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251, -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26112.875, 104448.875 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22848.791015625, 91392.8671875 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.10429885983467102, -0.10430017858743668, -0.09768327325582504, -0.09768451005220413, -0.08445210009813309, -0.08445316553115845, -0.110914446413517, -0.11091584712266922, -0.09106768667697906, -0.09106884151697159, -0.07783651351928711, -0.0778374969959259, -0.10429885983467102, -0.10430017858743668, -0.09768327325582504, -0.09768451005220413, -0.08445210009813309, -0.08445316553115845, -0.110914446413517, -0.11091584712266922, -0.09106768667697906, -0.09106884151697159, -0.07783651351928711, -0.0778374969959259 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162, -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26112.875, 104448.875 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22848.791015625, 91392.8671875 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.09355171024799347, 0.20322513580322266, 0.10347509384155273, 0.19826337695121765, 0.12332184612751007, 0.18833987414836884, 0.0836283266544342, 0.20818689465522766, 0.1133984625339508, 0.19330163300037384, 0.13324522972106934, 0.18337813019752502, 0.09355171024799347, 0.20322513580322266, 0.10347509384155273, 0.19826337695121765, 0.12332184612751007, 0.18833987414836884, 0.0836283266544342, 0.20818689465522766, 0.1133984625339508, 0.19330163300037384, 0.13324522972106934, 0.18337813019752502 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872, 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26738688.0, 106954752.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.46875, -1.46875, 0.29296875, 0.29296875, 1.46875, 1.46875 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23396352.0, 93585408.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "expected": [ -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375, -0.87890625, -0.87890625, -0.29296875, -0.29296875, 0.87890625, 0.87890625, -1.4609375, -1.4609375, 0.29296875, 0.29296875, 1.4609375, 1.4609375 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09298484772443771, -0.09298484772443771, -0.09277810901403427, -0.09277810901403427, -0.09236462414264679, -0.09236462414264679, -0.09319158643484116, -0.09319158643484116, -0.09257137030363083, -0.09257137030363083, -0.09215788543224335, -0.09215788543224335, -0.09298484772443771, -0.09298484772443771, -0.09277810901403427, -0.09277810901403427, -0.09236462414264679, -0.09236462414264679, -0.09319158643484116, -0.09319158643484116, -0.09257137030363083, -0.09257137030363083, -0.09215788543224335, -0.09215788543224335 ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26738688.0, 106954752.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.953125, 1.2265625, 0.689453125, -0.0947265625, 2.453125, -0.9765625 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23396352.0, 93585408.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "expected": [ -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125, -1.068359375, 0.7841796875, -0.189453125, 0.3447265625, 1.568359375, -0.5341796875, -1.94140625, 1.220703125, 0.689453125, -0.0947265625, 2.44140625, -0.970703125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.11052273213863373, 0.19473862648010254, 0.11083284020423889, 0.19458357989788055, 0.11145305633544922, 0.1942734718322754, 0.11021262407302856, 0.19489368796348572, 0.11114294826984406, 0.19442853331565857, 0.11176317930221558, 0.1941184103488922, 0.11052273213863373, 0.19473862648010254, 0.11083284020423889, 0.19458357989788055, 0.11145305633544922, 0.1942734718322754, 0.11021262407302856, 0.19489368796348572, 0.11114294826984406, 0.19442853331565857, 0.11176317930221558, 0.1941184103488922 ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474, 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474 ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 0.025, "atol": 2.5e-05 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.bfloat16/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0168359279632568, -1.0552423000335693, -0.2542089819908142, -0.26381057500839233, 1.2710447311401367, 1.3190526962280273 ], "expected": [ -1.0168358087539673, -1.0552421808242798, -0.2542089521884918, -0.26381054520606995, 1.2710447311401367, 1.319052815437317 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0168359279632568, -1.0552423000335693, -0.2542089819908142, -0.26381057500839233, 1.2710447311401367, 1.3190526962280273 ], "expected": [ -1.0168358087539673, -1.0552421808242798, -0.2542089521884918, -0.26381054520606995, 1.2710447311401367, 1.319052815437317 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0168359279632568, -1.0552423000335693, -0.2542089819908142, -0.26381057500839233, 1.2710447311401367, 1.3190526962280273 ], "expected": [ -1.0168358087539673, -1.0552421808242798, -0.2542089521884918, -0.26381054520606995, 1.2710447311401367, 1.319052815437317 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0168359279632568, -1.0552423000335693, -0.2542089819908142, -0.26381057500839233, 1.2710447311401367, 1.3190526962280273 ], "expected": [ -1.0168358087539673, -1.0552421808242798, -0.2542089521884918, -0.26381054520606995, 1.2710447311401367, 1.319052815437317 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.399752140045166, 2.7066047191619873 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.399752140045166, 2.7066047191619873 ], "expected": [ -1.9886621236801147, -3.84533953666687, -0.8591906428337097, -1.6613579988479614, 1.3997523784637451, 2.7066049575805664 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.2752538919448853, 0.916431725025177, -0.1313134729862213, 0.32285791635513306, 2.156567096710205, -0.8642895221710205 ], "expected": [ -1.275253713130951, 0.9164316356182098, -0.13131342828273773, 0.32285790890455246, 2.156567096710205, -0.8642896115779877 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.2752538919448853, 0.916431725025177, -0.1313134729862213, 0.32285791635513306, 2.156567096710205, -0.8642895221710205 ], "expected": [ -1.275253713130951, 0.9164316356182098, -0.13131342828273773, 0.32285790890455246, 2.156567096710205, -0.8642896115779877 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.2752538919448853, 0.916431725025177, -0.1313134729862213, 0.32285791635513306, 2.156567096710205, -0.8642895221710205 ], "expected": [ -1.275253713130951, 0.9164316356182098, -0.13131342828273773, 0.32285790890455246, 2.156567096710205, -0.8642896115779877 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.2752538919448853, 0.916431725025177, -0.1313134729862213, 0.32285791635513306, 2.156567096710205, -0.8642895221710205 ], "expected": [ -1.275253713130951, 0.9164316356182098, -0.13131342828273773, 0.32285790890455246, 2.156567096710205, -0.8642896115779877 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00032552084303461015, 0.0006510416860692203 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178217887878, 0.8750712275505066 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607015073299408, 0.5214030146598816 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838698029518127, 0.8386040329933167 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7329931259155273, 3.009004592895508, -1.0387859344482422, 1.3710185289382935, 2.349628210067749, -1.9049534797668457 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7329931259155273, 3.009004592895508, -1.0387859344482422, 1.3710185289382935, 2.349628210067749, -1.9049534797668457 ], "expected": [ -2.732993185520172, 3.0090046525001526, -1.0387859642505646, 1.371018499135971, 2.3496285676956177, -1.9049537181854248 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363016843795776, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363016843795776, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363016843795776, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363016843795776, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7627670764923096, -2.635500907897949, -0.78174889087677, -1.1687873601913452, 1.180287480354309, 1.7646397352218628 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7627670764923096, -2.635500907897949, -0.78174889087677, -1.1687873601913452, 1.180287480354309, 1.7646397352218628 ], "expected": [ -1.7627670764923096, -2.6355011463165283, -0.78174889087677, -1.1687874794006348, 1.1802875995635986, 1.7646398544311523 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.2544524669647217, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.2544524669647217, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.2544524669647217, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.2544524669647217, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2968750298023224, 0.5937500596046448 ], "expected": [ 0.296875, 0.59375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625, 1.859375 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.394150733947754, 2.101625680923462, -0.9226233959197998, 1.0015904903411865, 2.0204312801361084, -1.1984797716140747 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.394150733947754, 2.101625680923462, -0.9226233959197998, 1.0015904903411865, 2.0204312801361084, -1.1984797716140747 ], "expected": [ -2.3941506147384644, 2.1016258597373962, -0.922623336315155, 1.001590609550476, 2.020431399345398, -1.1984798908233643 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804812490940094, -0.06258240342140198, -0.06258348375558853 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804812490940094, -0.06258240342140198, -0.06258348375558853 ], "expected": [ -0.08577896654605865, -0.08578044176101685, -0.0780467763543129, -0.07804811745882034, -0.06258240342140198, -0.06258347630500793 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593750953674316, 19.187501907348633 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.1171875, 66902.171875 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.12133154273033142, 0.18933533132076263, 0.13292983174324036, 0.183536097407341, 0.15612639486789703, 0.17193761467933655 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.12133154273033142, 0.18933533132076263, 0.13292983174324036, 0.183536097407341, 0.15612639486789703, 0.17193761467933655 ], "expected": [ 0.12133155018091202, 0.18933533132076263, 0.13292983546853065, 0.18353608809411526, 0.15612639486789703, 0.17193760722875595 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690416097640991, -1.0690441131591797, -0.2672603726387024, -0.26726099848747253, 1.3363019227981567, 1.3363051414489746 ], "expected": [ -1.0690414905548096, -1.0690441131591797, -0.2672603726387024, -0.2672610282897949, 1.3363019227981567, 1.3363051414489746 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.336306095123291, 1.336306095123291 ], "expected": [ -1.0690449476242065, -1.0690449476242065, -0.26726123690605164, -0.26726123690605164, 1.3363062143325806, 1.3363062143325806 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689904689788818, -1.0690315961837769, -0.2672479748725891, -0.26725828647613525, 1.336236834526062, 1.336288332939148 ], "expected": [ -1.0689899921417236, -1.0690312385559082, -0.2672474980354309, -0.26725780963897705, 1.3362375497817993, 1.3362890481948853 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "expected": [ -0.07259416580200195, -0.07259416580200195, -0.07235252857208252, -0.07235252857208252, -0.07186925411224365, -0.07186925411224365 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.353562355041504, 0.9267830848693848, -0.1508905589580536, 0.3254457414150238, 2.25445294380188, -0.877228856086731 ], "expected": [ -1.3535622358322144, 0.9267830848693848, -0.1508905589580536, 0.3254457712173462, 2.254452884197235, -0.877228856086731 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.353567361831665, 0.9267836809158325, -0.15089184045791626, 0.32544592022895813, 2.2544591426849365, -0.8772295713424683 ], "expected": [ -1.3535674214363098, 0.9267837107181549, -0.15089185535907745, 0.3254459276795387, 2.254459321498871, -0.8772296607494354 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333435058594, 682.6666870117188 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573420.0, 78293680.0 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534857034683228, 0.9267736673355103, -0.15087196230888367, 0.32544371485710144, 2.2543551921844482, -0.8772162199020386 ], "expected": [ -1.3534849882125854, 0.9267734289169312, -0.15087124705314636, 0.3254433572292328, 2.254356324672699, -0.8772167861461639 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270935058594, 597.8541870117188 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.0, 68506968.0 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "expected": [ 0.14110875129699707, 0.17944562435150146, 0.14147120714187622, 0.1792643964290619, 0.14219611883163452, 0.17890194058418274 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9831992387771606, -3.7757554054260254, -0.8602430820465088, -1.6377917528152466, 1.385669231414795, 2.6381351947784424, -3.1061553955078125, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086252689361572, 4.776098728179932 ], "expected": [ -1.9831993579864502, -3.7757554054260254, -0.8602431416511536, -1.6377917528152466, 1.3856693506240845, 2.6381351947784424, -3.1061556339263916, -5.9137187004089355, 0.26271307468414307, 0.5001717805862427, 2.5086255073547363, 4.776098728179932 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7247989177703857, 2.9568166732788086, -1.0403646230697632, 1.3533438444137573, 2.3285038471221924, -1.8536014556884766, -4.409233093261719, 4.560288906097412, 0.6440696120262146, -0.250128835439682, 4.012938022613525, -3.4570741653442383 ], "expected": [ -2.7247990369796753, 2.956816554069519, -1.0403647124767303, 1.353343814611435, 2.3285040259361267, -1.8536013960838318, -4.409233450889587, 4.560289025306702, 0.6440696120262146, -0.250128835439682, 4.0129382610321045, -3.4570740461349487 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.6787298917770386, -2.346954345703125, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009504795074463, -3.636268377304077, 0.16571153700351715, 0.23167361319065094, 2.01015305519104, 2.8103017807006836 ], "expected": [ -1.6787300109863281, -2.346954584121704, -0.7565091848373413, -1.0576404333114624, 1.0879322290420532, 1.520987629890442, -2.6009507179260254, -3.6362686157226562, 0.16571153700351715, 0.23167362809181213, 2.01015305519104, 2.8103017807006836 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.268094778060913, 1.8852157592773438, -0.8847637176513672, 0.9182302951812744, 1.8818984031677246, -1.0157407522201538, -3.651425838470459, 2.852201223373413, 0.4985673129558563, -0.0487552136182785, 3.2652297019958496, -1.9827263355255127 ], "expected": [ -2.268095016479492, 1.885215938091278, -0.884763777256012, 0.9182303249835968, 1.8818983435630798, -1.0157407224178314, -3.651426076889038, 2.852201461791992, 0.4985673055052757, -0.0487552210688591, 3.26522958278656, -1.9827263355255127 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09953396022319794, -0.09953510761260986, -0.09322060644626617, -0.09322167932987213, -0.08059389889240265, -0.08059483021497726, -0.1058473140001297, -0.105848528444767, -0.08690725266933441, -0.0869082510471344, -0.07428054511547089, -0.07428140193223953 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09953396022319794, -0.09953510761260986, -0.09322060644626617, -0.09322167932987213, -0.08059389889240265, -0.08059483021497726, -0.1058473140001297, -0.105848528444767, -0.08690725266933441, -0.0869082510471344, -0.07428054511547089, -0.07428140193223953 ], "expected": [ -0.09953396767377853, -0.09953510761260986, -0.09322061389684677, -0.09322167932987213, -0.08059390634298325, -0.08059483021497726, -0.1058473214507103, -0.105848528444767, -0.08690726011991501, -0.0869082510471344, -0.07428055256605148, -0.07428140193223953 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.10069906711578369, 0.1996513307094574, 0.11016908288002014, 0.1949162632226944, 0.12910914421081543, 0.1854461133480072, 0.09122902154922485, 0.2043863981962204, 0.11963912844657898, 0.1901811957359314, 0.13857918977737427, 0.1807110607624054 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.10069906711578369, 0.1996513307094574, 0.11016908288002014, 0.1949162632226944, 0.12910914421081543, 0.1854461133480072, 0.09122902154922485, 0.2043863981962204, 0.11963912844657898, 0.1901811957359314, 0.13857918977737427, 0.1807110607624054 ], "expected": [ 0.1006990484893322, 0.1996513307094574, 0.11016907915472984, 0.1949162594974041, 0.12910914048552513, 0.18544612266123295, 0.09122901782393456, 0.20438639633357525, 0.11963910982012749, 0.1901811882853508, 0.13857917115092278, 0.18071105144917965 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893399685621262, -0.08893399685621262, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893399685621262, -0.08893399685621262, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "expected": [ -0.08873669803142548, -0.08873669803142548, -0.08853939920663834, -0.08853939920663834, -0.08814480900764465, -0.08814480900764465, -0.08893398940563202, -0.08893398940563202, -0.0883421078324318, -0.0883421078324318, -0.08794751763343811, -0.08794751763343811 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.0, 117440512.0 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.0, 102760448.0 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.11689496040344238, 0.1915525197982788, 0.1171908974647522, 0.1914045512676239, 0.11778278648853302, 0.1911086142063141, 0.11659900844097137, 0.19170048832893372, 0.11748683452606201, 0.191256582736969, 0.11807872354984283, 0.19096064567565918 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11689496040344238, 0.1915525197982788, 0.1171908974647522, 0.1914045512676239, 0.11778278648853302, 0.1911086142063141, 0.11659900844097137, 0.19170048832893372, 0.11748683452606201, 0.191256582736969, 0.11807872354984283, 0.19096064567565918 ], "expected": [ 0.11689495295286179, 0.1915525235235691, 0.1171909011900425, 0.19140454940497875, 0.11778278648853302, 0.1911086067557335, 0.11659901589155197, 0.19170049205422401, 0.11748683825135231, 0.19125658087432384, 0.11807872354984283, 0.19096063822507858 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.87220698595047, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.87220698595047, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "expected": [ -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947, -0.8546340465545654, -0.8722070455551147, -0.2848780155181885, -0.29073566198349, 0.8546340465545654, 0.8722070455551147, -1.4243900775909424, -1.4536783695220947, 0.2848780155181885, 0.29073566198349, 1.4243900775909424, 1.4536783695220947 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.3878480195999146, 2.6532649993896484, -3.111039400100708, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125699043273926, 4.803489685058594 ], "expected": [ -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594, -1.9863176345825195, -3.7974090576171875, -0.8615957498550415, -1.6471844911575317, 1.387848138809204, 2.6532649993896484, -3.111039638519287, -5.947633743286133, 0.2631261646747589, 0.5030402541160583, 2.5125701427459717, 4.803489685058594 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158, -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571, -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158, -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571, -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158, -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571, -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158, -1.0319510698318481, 0.7791552543640137, -0.17731702327728271, 0.3430517315864563, 1.5319510698318481, -0.5291552543640137, -1.8865852355957031, 1.2152588367462158, 0.6773170232772827, -0.09305174648761749, 2.386585235595703, -0.9652588367462158 ], "expected": [ -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571, -1.0319510698318481, 0.7791552841663361, -0.17731702327728271, 0.3430517464876175, 1.5319510698318481, -0.5291552841663361, -1.8865851163864136, 1.215258777141571, 0.6773170232772827, -0.09305174648761749, 2.3865851163864136, -0.965258777141571 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242590904236, 0.8750970959663391 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041575431824, 0.865141749382019 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.042393684387207, 1.3603883981704712, 2.3317720890045166, -1.8649487495422363, -4.416559219360352, 4.5857253074646, 0.6446892619132996, -0.25228017568588257, 4.018855094909668, -3.4776172637939453 ], "expected": [ -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453, -2.7294764518737793, 2.9730567932128906, -1.0423936247825623, 1.3603883683681488, 2.331772208213806, -1.8649487495422363, -4.416559457778931, 4.5857253074646, 0.6446892470121384, -0.25228019058704376, 4.0188552141189575, -3.4776172637939453 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668, -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668, -1.7059983015060425, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544875144958496, 0.1684032678604126, 0.23920556902885437, 2.042804718017578, 2.901667594909668 ], "expected": [ -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668, -1.705998420715332, -2.4232563972473145, -0.7687975168228149, -1.0920253992080688, 1.1056040525436401, 1.5704365968704224, -2.6431992053985596, -3.7544872760772705, 0.1684032678604126, 0.23920556902885437, 2.0428049564361572, 2.901667594909668 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272510528564, 2.465909004211426 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943723678589, 2.2571022510528564 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251, -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251, -2.308997392654419, 1.9424422979354858, -0.9031963348388672, 0.944019079208374, 1.9084060192108154, -1.0528274774551392, -3.714798927307129, 2.9408655166625977, 0.5026049017906189, -0.05440416932106018, 3.314207077026367, -2.051250696182251 ], "expected": [ -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251, -2.308997631072998, 1.9424422979354858, -0.9031962752342224, 0.9440190494060516, 1.9084060788154602, -1.0528274476528168, -3.7147988080978394, 2.940865457057953, 0.5026049017906189, -0.05440417677164078, 3.314207434654236, -2.051250696182251 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162, -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162, -0.10439193993806839, -0.10439325869083405, -0.09777044504880905, -0.09777168929576874, -0.08452747017145157, -0.08452853560447693, -0.11101342737674713, -0.11101483553647995, -0.09114895761013031, -0.09115011245012283, -0.07790597528219223, -0.07790696620941162 ], "expected": [ -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102, -0.10439194738864899, -0.10439325869083405, -0.09777045249938965, -0.09777168184518814, -0.08452747017145157, -0.08452853560447693, -0.11101343482732773, -0.11101483553647995, -0.09114896506071091, -0.09115011245012283, -0.07790598273277283, -0.07790695875883102 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.330078125, 104262.6953125 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.064453125, 91229.9609375 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872, 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872, 0.09341208636760712, 0.20329494774341583, 0.10334433615207672, 0.19832876324653625, 0.12320879101753235, 0.1883963942527771, 0.0834798514842987, 0.2082611322402954, 0.11327657103538513, 0.19336259365081787, 0.13314104080200195, 0.18343022465705872 ], "expected": [ 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327, 0.09341207891702652, 0.20329494401812553, 0.10334432125091553, 0.1983287613838911, 0.12320879474282265, 0.1883964017033577, 0.08347984775900841, 0.20826112665235996, 0.11327655240893364, 0.19336258433759212, 0.13314102590084076, 0.18343021906912327 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096075057983, -0.29276952147483826, -0.2927698791027069, 0.8783085346221924, 0.8783096075057983, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927698791027069, 1.4638476371765137, 1.463849425315857 ], "expected": [ -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857, -0.8783085346221924, -0.8783096671104431, -0.29276952147483826, -0.2927699089050293, 0.8783085346221924, 0.8783096671104431, -1.4638476371765137, -1.463849425315857, 0.29276952147483826, 0.2927699089050293, 1.4638476371765137, 1.463849425315857 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047, -0.878309965133667, -0.878309965133667, -0.29276999831199646, -0.29276999831199646, 0.878309965133667, 0.878309965133667, -1.4638500213623047, -1.4638500213623047, 0.29276999831199646, 0.29276999831199646, 1.4638500213623047, 1.4638500213623047 ], "expected": [ -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942, -0.8783100843429565, -0.8783100843429565, -0.29277002811431885, -0.29277002811431885, 0.8783100843429565, 0.8783100843429565, -1.4638501405715942, -1.4638501405715942, 0.29277002811431885, 0.29277002811431885, 1.4638501405715942, 1.4638501405715942 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824, -0.8782859444618225, -0.8783040046691895, -0.29276198148727417, -0.2927680015563965, 0.8782859444618225, 0.8783040046691895, -1.4638099670410156, -1.4638400077819824, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.4638400077819824 ], "expected": [ -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272, -0.8782860040664673, -0.8783040642738342, -0.29276198148727417, -0.2927680015563965, 0.8782860040664673, 0.8783040642738342, -1.4638099670410156, -1.463840126991272, 0.29276198148727417, 0.2927680015563965, 1.4638099670410156, 1.463840126991272 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "expected": [ -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279, -0.09306783229112625, -0.09306783229112625, -0.09286090731620789, -0.09286090731620789, -0.09244705736637115, -0.09244705736637115, -0.09327475726604462, -0.09327475726604462, -0.09265398234128952, -0.09265398234128952, -0.09224013239145279, -0.09224013239145279 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703, -1.0674628019332886, 0.7837321758270264, -0.1891542673110962, 0.3445774018764496, 1.5674628019332886, -0.5337321758270264, -1.9457714557647705, 1.2228870391845703, 0.6891542673110962, -0.09457740187644958, 2.4457714557647705, -0.9728870391845703 ], "expected": [ -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927, -1.0674628019332886, 0.7837322503328323, -0.18915428221225739, 0.344577431678772, 1.5674628019332886, -0.5337322503328323, -1.9457714557647705, 1.2228870689868927, 0.6891542822122574, -0.09457743167877197, 2.4457714557647705, -0.9728870689868927 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285, -1.0674649477005005, 0.7837324738502502, -0.1891549825668335, 0.34457749128341675, 1.5674649477005005, -0.5337324738502502, -1.945775032043457, 1.2228875160217285, 0.6891549825668335, -0.09457749128341675, 2.445775032043457, -0.9728875160217285 ], "expected": [ -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957, -1.0674651265144348, 0.7837325632572174, -0.18915504217147827, 0.34457752108573914, 1.5674651265144348, -0.5337325632572174, -1.9457752108573914, 1.2228876054286957, 0.6891550421714783, -0.09457752108573914, 2.4457752108573914, -0.9728876054286957 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.0, 106764104.0 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868, -1.0674289464950562, 0.7837280035018921, -0.18914297223091125, 0.34457600116729736, 1.5674289464950562, -0.5337280035018921, -1.9457149505615234, 1.2228800058364868, 0.6891429424285889, -0.09457600116729736, 2.4457149505615234, -0.9728800058364868 ], "expected": [ -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454, -1.067429006099701, 0.7837280482053757, -0.18914297223091125, 0.34457600116729736, 1.567429006099701, -0.5337280482053757, -1.9457149505615234, 1.222880095243454, 0.6891429722309113, -0.09457600116729736, 2.4457149505615234, -0.972880095243454 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.0, 93418592.0 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474, 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474 ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474, 0.11039824783802032, 0.19480088353157043, 0.11070863902568817, 0.19464567303657532, 0.11132940649986267, 0.19433529675006866, 0.11008787155151367, 0.19495606422424316, 0.11101903021335602, 0.1944904923439026, 0.11163979768753052, 0.19418010115623474 ], "expected": [ 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896, 0.11039825156331062, 0.1948008742183447, 0.11070863902568817, 0.19464568048715591, 0.11132941395044327, 0.19433529302477837, 0.11008786410093307, 0.19495606794953346, 0.11101902648806572, 0.19449048675596714, 0.11163980141282082, 0.1941800992935896 ], "rtol": 6e-06, "atol": 6e-09 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float32/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.016835820076726, -1.0552422211906038, -0.2542089550191815, -0.26381055529765096, 1.2710447750959077, 1.319052776488255 ], "expected": [ -1.016835820076726, -1.0552422211906036, -0.2542089550191815, -0.2638105552976509, 1.2710447750959077, 1.3190527764882547 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.016835820076726, -1.0552422211906038, -0.2542089550191815, -0.26381055529765096, 1.2710447750959077, 1.319052776488255 ], "expected": [ -1.016835820076726, -1.0552422211906036, -0.2542089550191815, -0.2638105552976509, 1.2710447750959077, 1.3190527764882547 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.016835820076726, -1.0552422211906038, -0.2542089550191815, -0.26381055529765096, 1.2710447750959077, 1.319052776488255 ], "expected": [ -1.016835820076726, -1.0552422211906036, -0.2542089550191815, -0.2638105552976509, 1.2710447750959077, 1.3190527764882547 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003255208333333333, 0.0006510416666666666 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178019205729, 0.8750712076822916 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.016835820076726, -1.0552422211906038, -0.2542089550191815, -0.26381055529765096, 1.2710447750959077, 1.319052776488255 ], "expected": [ -1.016835820076726, -1.0552422211906036, -0.2542089550191815, -0.2638105552976509, 1.2710447750959077, 1.3190527764882547 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0003255208333333333, 0.0006510416666666666 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178019205729, 0.8750712076822916 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2607014973958333, 0.5214029947916666 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838697433471679, 0.8386039733886718 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607014973958333, 0.5214029947916666 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838697433471679, 0.8386039733886718 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9886621718329953, -3.845339436664547, -0.859190665859764, -1.6613579711225464, 1.399752346086699, 2.7066049599614552 ], "expected": [ -1.9886621718329958, -3.8453394366645473, -0.8591906658597642, -1.6613579711225464, 1.3997523460866987, 2.7066049599614543 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9886621718329953, -3.845339436664547, -0.859190665859764, -1.6613579711225464, 1.399752346086699, 2.7066049599614552 ], "expected": [ -1.9886621718329958, -3.8453394366645473, -0.8591906658597642, -1.6613579711225464, 1.3997523460866987, 2.7066049599614543 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.275253730115089, 0.9164316658929529, -0.13131343252877226, 0.32285791647323825, 2.156567162643862, -0.8642895823661911 ], "expected": [ -1.275253730115089, 0.9164316658929528, -0.13131343252877226, 0.3228579164732382, 2.156567162643862, -0.864289582366191 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.275253730115089, 0.9164316658929529, -0.13131343252877226, 0.32285791647323825, 2.156567162643862, -0.8642895823661911 ], "expected": [ -1.275253730115089, 0.9164316658929528, -0.13131343252877226, 0.3228579164732382, 2.156567162643862, -0.864289582366191 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.275253730115089, 0.9164316658929529, -0.13131343252877226, 0.32285791647323825, 2.156567162643862, -0.8642895823661911 ], "expected": [ -1.275253730115089, 0.9164316658929528, -0.13131343252877226, 0.3228579164732382, 2.156567162643862, -0.864289582366191 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0003255208333333333, 0.0006510416666666666 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750178019205729, 0.8750712076822916 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.275253730115089, 0.9164316658929529, -0.13131343252877226, 0.32285791647323825, 2.156567162643862, -0.8642895823661911 ], "expected": [ -1.275253730115089, 0.9164316658929528, -0.13131343252877226, 0.3228579164732382, 2.156567162643862, -0.864289582366191 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0003255208333333333, 0.0006510416666666666 ], "expected": [ 0.0003255208333333333, 0.0006510416666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750178019205729, 0.8750712076822916 ], "expected": [ 0.8750178019205729, 0.8750712076822916 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.2607014973958333, 0.5214029947916666 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7838697433471679, 0.8386039733886718 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.2607014973958333, 0.5214029947916666 ], "expected": [ 0.26070149739583337, 0.5214029947916667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7838697433471679, 0.8386039733886718 ], "expected": [ 0.7838697433471679, 0.8386039733886718 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.732993257749493, 3.00900457749841, -1.038785998789646, 1.3710184783419097, 2.3496285191300483, -1.9049537199710915 ], "expected": [ -2.7329932577494938, 3.0090045774984104, -1.0387859987896464, 1.3710184783419097, 2.3496285191300483, -1.9049537199710906 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.732993257749493, 3.00900457749841, -1.038785998789646, 1.3710184783419097, 2.3496285191300483, -1.9049537199710915 ], "expected": [ -2.7329932577494938, 3.0090045774984104, -1.0387859987896464, 1.3710184783419097, 2.3496285191300483, -1.9049537199710906 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.2672603828625743, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.2672603828625743, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.2672603828625743, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.2672603828625743, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.29687499999999994, 0.5937499999999999 ], "expected": [ 0.296875, 0.59375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.0390625000000002, 1.8593750000000002 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.29687499999999994, 0.5937499999999999 ], "expected": [ 0.296875, 0.59375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625000000002, 1.8593750000000002 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7627671185512954, -2.635501091903876, -0.7817488960531832, -1.1687874407573713, 1.1802875489430413, 1.764639861535639 ], "expected": [ -1.7627671185512954, -2.635501091903876, -0.7817488960531832, -1.1687874407573713, 1.1802875489430413, 1.764639861535639 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7627671185512954, -2.635501091903876, -0.7817488960531832, -1.1687874407573713, 1.1802875489430413, 1.764639861535639 ], "expected": [ -1.7627671185512954, -2.635501091903876, -0.7817488960531832, -1.1687874407573713, 1.1802875489430413, 1.764639861535639 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386143, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386143, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386143, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386143, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.29687499999999994, 0.5937499999999999 ], "expected": [ 0.296875, 0.59375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.0390625000000002, 1.8593750000000002 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.29687499999999994, 0.5937499999999999 ], "expected": [ 0.296875, 0.59375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.0390625000000002, 1.8593750000000002 ], "expected": [ 1.0390625000000002, 1.8593750000000002 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.3941506778269432, 2.101625818927907, -0.9226233440797749, 1.0015905805680285, 2.020431323414562, -1.1984798961517291 ], "expected": [ -2.3941506778269432, 2.101625818927907, -0.9226233440797749, 1.0015905805680285, 2.020431323414562, -1.1984798961517291 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.3941506778269432, 2.101625818927907, -0.9226233440797749, 1.0015905805680285, 2.020431323414562, -1.1984798961517291 ], "expected": [ -2.3941506778269432, 2.101625818927907, -0.9226233440797749, 1.0015905805680285, 2.020431323414562, -1.1984798961517291 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "expected": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "expected": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "expected": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "expected": [ -1.069044967597265, -1.0690449676365894, -0.26726124189931627, -0.26726124190914735, 1.3363062094965814, 1.3363062095457368 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.593749999999998, 19.187499999999996 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16726.117187500004, 66902.17187500001 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593749999999998, 19.187499999999996 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.117187500004, 66902.17187500001 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.08577896602765822, -0.0857804385093382, -0.07804677754065804, -0.07804811729159504, -0.06258240056665769, -0.06258347485610871 ], "expected": [ -0.08577896602765822, -0.0857804385093382, -0.07804677754065804, -0.07804811729159504, -0.06258240056665769, -0.06258347485610871 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.08577896602765822, -0.0857804385093382, -0.07804677754065804, -0.07804811729159504, -0.06258240056665769, -0.06258347485610871 ], "expected": [ -0.08577896602765822, -0.0857804385093382, -0.07804677754065804, -0.07804811729159504, -0.06258240056665769, -0.06258347485610871 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "expected": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "expected": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "expected": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "expected": [ -1.3535674513958975, 0.926783725727442, -0.15089186284897438, 0.3254459314318605, 2.254459314244872, -0.8772296571593026 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 9.593749999999998, 19.187499999999996 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 16726.117187500004, 66902.17187500001 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 9.593749999999998, 19.187499999999996 ], "expected": [ 9.593749999999998, 19.187499999999996 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 16726.117187500004, 66902.17187500001 ], "expected": [ 16726.117187500004, 66902.17187500001 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.12133155095851267, 0.18933532888200366, 0.13292983368901293, 0.18353608796869628, 0.15612639915001347, 0.17193760614208153 ], "expected": [ 0.12133155095851267, 0.18933532888200366, 0.13292983368901293, 0.18353608796869628, 0.15612639915001347, 0.17193760614208153 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.12133155095851267, 0.18933532888200366, 0.13292983368901293, 0.18353608796869628, 0.15612639915001347, 0.17193760614208153 ], "expected": [ 0.12133155095851267, 0.18933532888200366, 0.13292983368901293, 0.18353608796869628, 0.15612639915001347, 0.17193760614208153 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.336306209562058, 1.3363062095621059 ], "expected": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.3363062095620581, 1.3363062095621059 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.336306209562058, 1.3363062095621059 ], "expected": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.3363062095620581, 1.3363062095621059 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.0690415314502975, -1.0690441085967413, -0.2672603828625744, -0.26726102714918537, 1.3363019143128718, 1.3363051357459266 ], "expected": [ -1.0690415314502972, -1.0690441085967413, -0.26726038286257436, -0.2672610271491853, 1.3363019143128718, 1.3363051357459266 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.336306209562058, 1.3363062095621059 ], "expected": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.3363062095620581, 1.3363062095621059 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.3333333333333, 682.6666666666666 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19573419.541666668, 78293675.54166667 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.336306209562058, 1.3363062095621059 ], "expected": [ -1.0690449676496463, -1.0690449676496847, -0.2672612419124116, -0.2672612419124212, 1.3363062095620581, 1.3363062095621059 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333333333333, 682.6666666666666 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573419.541666668, 78293675.54166667 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.9270833333333, 597.8541666666666 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17126742.117187504, 68506966.17187501 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0689899924351263, -1.0690312230509007, -0.2672474981087809, -0.2672578057627245, 1.33623749054391, 1.3362890288136282 ], "expected": [ -1.0689899924351276, -1.069031223050902, -0.26724749810878223, -0.26725780576272584, 1.3362374905439085, 1.3362890288136264 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270833333333, 597.8541666666666 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.117187504, 68506966.17187501 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.07259416300155497, -0.07259416421852534, -0.07235252645468943, -0.072352527667609, -0.07186925336095833, -0.0718692545657763 ], "expected": [ -0.07259416300155497, -0.07259416421852534, -0.07235252645468943, -0.072352527667609, -0.07186925336095833, -0.0718692545657763 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.07259416300155497, -0.07259416421852534, -0.07235252645468943, -0.072352527667609, -0.07186925336095833, -0.0718692545657763 ], "expected": [ -0.07259416300155497, -0.07259416421852534, -0.07235252645468943, -0.072352527667609, -0.07186925336095833, -0.0718692545657763 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.254459314343087, -0.8772296571715794 ], "expected": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.2544593143430873, -0.8772296571715794 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.254459314343087, -0.8772296571715794 ], "expected": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.2544593143430873, -0.8772296571715794 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.3535622971754462, 0.926783081447556, -0.15089057429386166, 0.32544577036188904, 2.254452871469308, -0.8772288518094449 ], "expected": [ -1.3535622971754457, 0.926783081447556, -0.15089057429386155, 0.325445770361889, 2.254452871469308, -0.8772288518094449 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.254459314343087, -0.8772296571715794 ], "expected": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.2544593143430873, -0.8772296571715794 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 341.3333333333333, 682.6666666666666 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 19573419.541666668, 78293675.54166667 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.254459314343087, -0.8772296571715794 ], "expected": [ -1.3535674514744693, 0.9267837257372635, -0.15089186286861733, 0.3254459314343159, 2.2544593143430873, -0.8772296571715794 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 341.3333333333333, 682.6666666666666 ], "expected": [ 341.3333333333333, 682.6666666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 19573419.541666668, 78293675.54166667 ], "expected": [ 19573419.541666668, 78293675.54166667 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 298.9270833333333, 597.8541666666666 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 17126742.117187504, 68506966.17187501 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.3534849886526894, 0.9267734172881755, -0.15087124716317135, 0.3254433543220434, 2.254356235815865, -0.8772167716102213 ], "expected": [ -1.3534849886526914, 0.9267734172881765, -0.15087124716317335, 0.3254433543220444, 2.254356235815863, -0.8772167716102199 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 298.9270833333333, 597.8541666666666 ], "expected": [ 298.9270833333333, 597.8541666666666 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 17126742.117187504, 68506966.17187501 ], "expected": [ 17126742.117187504, 68506966.17187501 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.14110875549766755, 0.179445623163894, 0.14147121031796586, 0.17926439575070674, 0.14219611995856252, 0.1789019409243322 ], "expected": [ 0.14110875549766755, 0.179445623163894, 0.14147121031796586, 0.17926439575070674, 0.14219611995856252, 0.1789019409243322 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.14110875549766755, 0.179445623163894, 0.14147121031796586, 0.17926439575070674, 0.14219611995856252, 0.1789019409243322 ], "expected": [ 0.14110875549766755, 0.179445623163894, 0.14147121031796586, 0.17926439575070674, 0.14219611995856252, 0.1789019409243322 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9831993644810715, -3.7757553163506214, -0.8602431388837715, -1.6377917738212078, 1.3856693123108286, 2.638135311237619, -3.1061555900783717, -5.913718858880035, 0.2627130867135285, 0.5001717687082057, 2.5086255379081286, 4.776098853767032 ], "expected": [ -1.9831993644810717, -3.7757553163506214, -0.8602431388837716, -1.6377917738212078, 1.3856693123108286, 2.6381353112376194, -3.1061555900783717, -5.913718858880035, 0.2627130867135285, 0.5001717687082057, 2.5086255379081286, 4.776098853767033 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9831993644810715, -3.7757553163506214, -0.8602431388837715, -1.6377917738212078, 1.3856693123108286, 2.638135311237619, -3.1061555900783717, -5.913718858880035, 0.2627130867135285, 0.5001717687082057, 2.5086255379081286, 4.776098853767032 ], "expected": [ -1.9831993644810717, -3.7757553163506214, -0.8602431388837716, -1.6377917738212078, 1.3856693123108286, 2.6381353112376194, -3.1061555900783717, -5.913718858880035, 0.2627130867135285, 0.5001717687082057, 2.5086255379081286, 4.776098853767033 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750267028808594, 0.8751068115234375 ], "expected": [ 0.8750267028808594, 0.8751068115234375 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.792992115020752, 0.8750934600830078 ], "expected": [ 0.792992115020752, 0.8750934600830078 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.724799046721607, 2.956816487262966, -1.040364708325657, 1.3533438303659058, 2.3285039684662427, -1.8536014834282142, -4.409233385117558, 4.560289144160026, 0.6440696300702927, -0.25012882653115426, 4.012938306862193, -3.4570741403252745 ], "expected": [ -2.7247990467216074, 2.956816487262966, -1.0403647083256573, 1.3533438303659058, 2.3285039684662427, -1.8536014834282146, -4.409233385117558, 4.560289144160026, 0.6440696300702927, -0.25012882653115426, 4.012938306862193, -3.457074140325275 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.724799046721607, 2.956816487262966, -1.040364708325657, 1.3533438303659058, 2.3285039684662427, -1.8536014834282142, -4.409233385117558, 4.560289144160026, 0.6440696300702927, -0.25012882653115426, 4.012938306862193, -3.4570741403252745 ], "expected": [ -2.7247990467216074, 2.956816487262966, -1.0403647083256573, 1.3533438303659058, 2.3285039684662427, -1.8536014834282146, -4.409233385117558, 4.560289144160026, 0.6440696300702927, -0.25012882653115426, 4.012938306862193, -3.457074140325275 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.6787299527935815, -2.346954505940403, -0.7565092061945324, -1.0576404425911687, 1.0879322870035657, 1.5209876841073, -2.6009506993926306, -3.6362685692896375, 0.16571154040451663, 0.23167362075806555, 2.0101530336026148, 2.8103017474565344 ], "expected": [ -1.6787299527935815, -2.346954505940403, -0.7565092061945324, -1.0576404425911687, 1.0879322870035657, 1.5209876841072998, -2.6009506993926306, -3.6362685692896375, 0.16571154040451663, 0.23167362075806555, 2.0101530336026148, 2.8103017474565344 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.6787299527935815, -2.346954505940403, -0.7565092061945324, -1.0576404425911687, 1.0879322870035657, 1.5209876841073, -2.6009506993926306, -3.6362685692896375, 0.16571154040451663, 0.23167362075806555, 2.0101530336026148, 2.8103017474565344 ], "expected": [ -1.6787299527935815, -2.346954505940403, -0.7565092061945324, -1.0576404425911687, 1.0879322870035657, 1.5209876841072998, -2.6009506993926306, -3.6362685692896375, 0.16571154040451663, 0.23167362075806555, 2.0101530336026148, 2.8103017474565344 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.3125, 2.625 ], "expected": [ 1.3125, 2.625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.17578125, 2.40625 ], "expected": [ 1.17578125, 2.40625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.2680949291903723, 1.8852158794553024, -0.8847638092917987, 0.9182303319433766, 1.8818984305053486, -1.015740763080475, -3.651426049088946, 2.8522014269672282, 0.49856731060677495, -0.04875521556854917, 3.265229550403922, -1.9827263105924007 ], "expected": [ -2.2680949291903723, 1.8852158794553024, -0.8847638092917987, 0.9182303319433766, 1.8818984305053486, -1.015740763080475, -3.651426049088946, 2.8522014269672282, 0.49856731060677495, -0.04875521556854917, 3.265229550403922, -1.9827263105924007 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.2680949291903723, 1.8852158794553024, -0.8847638092917987, 0.9182303319433766, 1.8818984305053486, -1.015740763080475, -3.651426049088946, 2.8522014269672282, 0.49856731060677495, -0.04875521556854917, 3.265229550403922, -1.9827263105924007 ], "expected": [ -2.2680949291903723, 1.8852158794553024, -0.8847638092917987, 0.9182303319433766, 1.8818984305053486, -1.015740763080475, -3.651426049088946, 2.8522014269672282, 0.49856731060677495, -0.04875521556854917, 3.265229550403922, -1.9827263105924007 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09953396579688467, -0.09953510487115394, -0.09322061216853916, -0.09322167899230968, -0.08059390491184815, -0.08059482723462118, -0.10584731942523018, -0.10584853074999818, -0.08690725854019365, -0.08690825311346542, -0.07428055128350262, -0.07428140135577692 ], "expected": [ -0.09953396579688467, -0.09953510487115394, -0.09322061216853916, -0.09322167899230968, -0.08059390491184813, -0.08059482723462118, -0.10584731942523018, -0.1058485307499982, -0.08690725854019364, -0.08690825311346544, -0.07428055128350262, -0.07428140135577692 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09953396579688467, -0.09953510487115394, -0.09322061216853916, -0.09322167899230968, -0.08059390491184815, -0.08059482723462118, -0.10584731942523018, -0.10584853074999818, -0.08690725854019365, -0.08690825311346542, -0.07428055128350262, -0.07428140135577692 ], "expected": [ -0.09953396579688467, -0.09953510487115394, -0.09322061216853916, -0.09322167899230968, -0.08059390491184813, -0.08059482723462118, -0.10584731942523018, -0.1058485307499982, -0.08690725854019364, -0.08690825311346544, -0.07428055128350262, -0.07428140135577692 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 28672.875, 114688.875 ], "expected": [ 28672.875, 114688.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25088.79296875, 100352.875 ], "expected": [ 25088.79296875, 100352.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.10069905130467299, 0.19965132865336546, 0.11016908174719126, 0.19491625924423225, 0.12910914263222778, 0.18544612042596587, 0.09122902086215473, 0.20438639806249864, 0.11963911218970952, 0.19018118983509907, 0.13857917307474607, 0.18071105101683269 ], "expected": [ 0.10069905130467299, 0.19965132865336546, 0.11016908174719126, 0.19491625924423225, 0.1291091426322278, 0.18544612042596587, 0.09122902086215473, 0.20438639806249864, 0.11963911218970955, 0.19018118983509907, 0.13857917307474607, 0.18071105101683269 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.10069905130467299, 0.19965132865336546, 0.11016908174719126, 0.19491625924423225, 0.12910914263222778, 0.18544612042596587, 0.09122902086215473, 0.20438639806249864, 0.11963911218970952, 0.19018118983509907, 0.13857917307474607, 0.18071105101683269 ], "expected": [ 0.10069905130467299, 0.19965132865336546, 0.11016908174719126, 0.19491625924423225, 0.1291091426322278, 0.18544612042596587, 0.09122902086215473, 0.20438639806249864, 0.11963911218970955, 0.19018118983509907, 0.13857917307474607, 0.18071105101683269 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.875, 117440512.875 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.875, 117440512.875 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.79296875, 102760448.875 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.79296875, 102760448.875 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.088736696002718, -0.0887366969944408, -0.0885394005869051, -0.08853940157642293, -0.08814480975527933, -0.0881448107403872, -0.08893399141853088, -0.08893399241245865, -0.08834210517109221, -0.08834210615840507, -0.08794751433946645, -0.08794751532236933 ], "expected": [ -0.088736696002718, -0.0887366969944408, -0.0885394005869051, -0.08853940157642293, -0.08814480975527933, -0.0881448107403872, -0.08893399141853088, -0.08893399241245865, -0.08834210517109221, -0.08834210615840507, -0.08794751433946645, -0.08794751532236934 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.088736696002718, -0.0887366969944408, -0.0885394005869051, -0.08853940157642293, -0.08814480975527933, -0.0881448107403872, -0.08893399141853088, -0.08893399241245865, -0.08834210517109221, -0.08834210615840507, -0.08794751433946645, -0.08794751532236933 ], "expected": [ -0.088736696002718, -0.0887366969944408, -0.0885394005869051, -0.08853940157642293, -0.08814480975527933, -0.0881448107403872, -0.08893399141853088, -0.08893399241245865, -0.08834210517109221, -0.08834210615840507, -0.08794751433946645, -0.08794751532236934 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 29360128.875, 117440512.875 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 29360128.875, 117440512.875 ], "expected": [ 29360128.875, 117440512.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 25690112.79296875, 102760448.875 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 25690112.79296875, 102760448.875 ], "expected": [ 25690112.79296875, 102760448.875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.116894955995923, 0.1915525227458306, 0.11719089911964237, 0.1914045511823172, 0.117782785367081, 0.1911086080552904, 0.11659901287220367, 0.191700494309344, 0.11748684224336167, 0.1912565796188038, 0.11807872849080034, 0.190960636491777 ], "expected": [ 0.116894955995923, 0.1915525227458306, 0.11719089911964237, 0.1914045511823172, 0.117782785367081, 0.1911086080552904, 0.11659901287220367, 0.191700494309344, 0.11748684224336167, 0.1912565796188038, 0.11807872849080034, 0.19096063649177702 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.116894955995923, 0.1915525227458306, 0.11719089911964237, 0.1914045511823172, 0.117782785367081, 0.1911086080552904, 0.11659901287220367, 0.191700494309344, 0.11748684224336167, 0.1912565796188038, 0.11807872849080034, 0.190960636491777 ], "expected": [ 0.116894955995923, 0.1915525227458306, 0.11719089911964237, 0.1914045511823172, 0.117782785367081, 0.1911086080552904, 0.11659901287220367, 0.191700494309344, 0.11748684224336167, 0.1912565796188038, 0.11807872849080034, 0.19096063649177702 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242753462358, 0.8750971013849432 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817947, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817947, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "expected": [ -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244, -0.8546340317639081, -0.8722070292817946, -0.28487801058796935, -0.2907356764272649, 0.8546340317639081, 0.8722070292817946, -1.4243900529398468, -1.4536783821363244, 0.28487801058796935, 0.2907356764272649, 1.4243900529398468, 1.4536783821363244 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242753462358, 0.8750971013849432 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905041954734109, 0.8651417818936435 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041954734109, 0.8651417818936435 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.9863176920624128, -3.797409126015789, -0.8615957612951561, -1.6471844458484246, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189396, 2.512570031006614, 4.803489594653668, -1.9863176920624128, -3.797409126015789, -0.8615957612951561, -1.6471844458484246, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189396, 2.512570031006614, 4.803489594653668 ], "expected": [ -1.9863176920624128, -3.7974091260157894, -0.8615957612951561, -1.6471844458484248, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189397, 2.512570031006614, 4.803489594653668, -1.9863176920624128, -3.7974091260157894, -0.8615957612951561, -1.6471844458484248, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189397, 2.512570031006614, 4.803489594653668 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.9863176920624128, -3.797409126015789, -0.8615957612951561, -1.6471844458484246, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189396, 2.512570031006614, 4.803489594653668, -1.9863176920624128, -3.797409126015789, -0.8615957612951561, -1.6471844458484246, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189396, 2.512570031006614, 4.803489594653668 ], "expected": [ -1.9863176920624128, -3.7974091260157894, -0.8615957612951561, -1.6471844458484248, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189397, 2.512570031006614, 4.803489594653668, -1.9863176920624128, -3.7974091260157894, -0.8615957612951561, -1.6471844458484248, 1.3878481002393572, 2.653264914486304, -3.1110396228296695, -5.947633806183154, 0.2631261694721005, 0.5030402343189397, 2.512570031006614, 4.803489594653668 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 0.8750242753462358, 0.8750971013849432 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.779155271961346, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.529155271961346, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "expected": [ -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433, -1.0319510476458622, 0.7791552719613459, -0.17731701588195403, 0.3430517573204487, 1.5319510476458622, -0.5291552719613459, -1.88658507940977, 1.2152587866022433, 0.677317015881954, -0.09305175732044868, 2.38658507940977, -0.9652587866022433 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.00048828125, 0.0009765625 ], "expected": [ 0.00048828125, 0.0009765625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 0.8750242753462358, 0.8750971013849432 ], "expected": [ 0.8750242753462358, 0.8750971013849432 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 0.7905041954734109, 0.8651417818936435 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.26605224609375, 0.5321044921875 ], "expected": [ 0.26605224609375, 0.5321044921875 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 0.7905041954734109, 0.8651417818936435 ], "expected": [ 0.7905041954734109, 0.8651417818936435 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.7294765380936195, 2.9730568445118415, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392047, 4.018855046509921, -3.477617195990251, -2.7294765380936195, 2.9730568445118415, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392047, 4.018855046509921, -3.477617195990251 ], "expected": [ -2.7294765380936195, 2.973056844511842, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392048, 4.018855046509921, -3.477617195990251, -2.7294765380936195, 2.973056844511842, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392048, 4.018855046509921, -3.477617195990251 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.7294765380936195, 2.9730568445118415, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392047, 4.018855046509921, -3.477617195990251, -2.7294765380936195, 2.9730568445118415, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392047, 4.018855046509921, -3.477617195990251 ], "expected": [ -2.7294765380936195, 2.973056844511842, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392048, 4.018855046509921, -3.477617195990251, -2.7294765380936195, 2.973056844511842, -1.0423936419427342, 1.3603883343863186, 2.331772150359036, -1.8649486858647282, -4.416559434244505, 4.585725354637365, 0.6446892542081507, -0.2522801757392048, 4.018855046509921, -3.477617195990251 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a0.0078125/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2727272727272727, 2.465909090909091 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272727272727, 2.465909090909091 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384943181818181, 2.257102272727273 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943181818181, 2.257102272727273 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -1.7059983637500167, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815616, 0.23920556602415005, 2.042804907666329, 2.9016675182929506, -1.7059983637500167, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815616, 0.23920556602415005, 2.042804907666329, 2.9016675182929506 ], "expected": [ -1.7059983637500165, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815613, 0.23920556602415005, 2.0428049076663286, 2.9016675182929506, -1.7059983637500165, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815613, 0.23920556602415005, 2.0428049076663286, 2.9016675182929506 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -1.7059983637500167, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815616, 0.23920556602415005, 2.042804907666329, 2.9016675182929506, -1.7059983637500167, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815616, 0.23920556602415005, 2.042804907666329, 2.9016675182929506 ], "expected": [ -1.7059983637500165, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815613, 0.23920556602415005, 2.0428049076663286, 2.9016675182929506, -1.7059983637500165, -2.423256386244651, -0.7687975458959302, -1.0920254101102502, 1.1056040898122426, 1.5704365421585504, -2.643199181604103, -3.754487362379051, 0.16840327195815613, 0.23920556602415005, 2.0428049076663286, 2.9016675182929506 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 1.2727272727272727, 2.465909090909091 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 0.0625, 0.125 ], "expected": [ 0.0625, 0.125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 1.2727272727272727, 2.465909090909091 ], "expected": [ 1.2727272727272727, 2.465909090909091 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 1.1384943181818181, 2.257102272727273 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 0.3203125, 0.640625 ], "expected": [ 0.3203125, 0.640625 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 1.1384943181818181, 2.257102272727273 ], "expected": [ 1.1384943181818181, 2.257102272727273 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.3142073614994936, -2.051250638719713, -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.3142073614994936, -2.051250638719713 ], "expected": [ -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.314207361499493, -2.051250638719713, -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.314207361499493, -2.051250638719713 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.3142073614994936, -2.051250638719713, -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.3142073614994936, -2.051250638719713 ], "expected": [ -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.314207361499493, -2.051250638719713, -2.308997545625025, 1.942442289683488, -0.9031963188438952, 0.9440190575826877, 1.9084061347183638, -1.052827406618913, -3.714798772406154, 2.9408655217842883, 0.5026049079372342, -0.054404174518112525, 3.314207361499493, -2.051250638719713 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a1.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272, -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227, -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272, -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227, -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272, -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227, -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26066.329545454544, 104262.69318181818 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272, -0.8783100656307051, -0.8783100656479363, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479363, -1.4638501093845084, -1.4638501094132272, 0.2927700218769017, 0.29277002188264545, 1.4638501093845084, 1.4638501094132272 ], "expected": [ -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227, -0.8783100656307051, -0.8783100656479362, -0.2927700218769017, -0.29277002188264545, 0.8783100656307051, 0.8783100656479362, -1.4638501093845087, -1.463850109413227, 0.2927700218769017, 0.29277002188264545, 1.4638501093845087, 1.463850109413227 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.329545454544, 104262.69318181818 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22808.063210227272, 91229.95596590909 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.063210227272, 91229.95596590909 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.10439194473484711, -0.10439325886937698, -0.09777045369120964, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654, -0.10439194473484711, -0.10439325886937698, -0.09777045369120964, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654 ], "expected": [ -0.10439194473484713, -0.10439325886937698, -0.09777045369120965, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654, -0.10439194473484713, -0.10439325886937698, -0.09777045369120965, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.10439194473484711, -0.10439325886937698, -0.09777045369120964, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654, -0.10439194473484711, -0.10439325886937698, -0.09777045369120964, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654 ], "expected": [ -0.10439194473484713, -0.10439325886937698, -0.09777045369120965, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654, -0.10439194473484713, -0.10439325886937698, -0.09777045369120965, -0.09777168447131936, -0.08452747160393469, -0.08452853567520414, -0.1110134357784846, -0.11101483326743458, -0.09114896264757216, -0.09115011007326175, -0.07790598056029721, -0.07790696127714654 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203, -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203, -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203, -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203, -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203, -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203, -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26066.329545454544, 104262.69318181818 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203, -1.0674650984460576, 0.7837325492359523, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359523, -1.9457751640767627, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.4457751640767627, -0.9728875820599203 ], "expected": [ -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203, -1.0674650984460576, 0.7837325492359521, -0.18915503281535256, 0.3445775164119841, 1.5674650984460576, -0.5337325492359521, -1.945775164076763, 1.2228875820599203, 0.6891550328153526, -0.09457751641198409, 2.445775164076763, -0.9728875820599203 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 16.0, 32.0 ], "expected": [ 16.0, 32.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26066.329545454544, 104262.69318181818 ], "expected": [ 26066.329545454544, 104262.69318181818 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 22808.063210227272, 91229.95596590909 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 14.265625, 28.53125 ], "expected": [ 14.265625, 28.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 22808.063210227272, 91229.95596590909 ], "expected": [ 22808.063210227272, 91229.95596590909 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.09341208289772934, 0.20329494415203272, 0.10334431946318554, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599, 0.09341208289772934, 0.20329494415203272, 0.10334431946318554, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599 ], "expected": [ 0.09341208289772931, 0.20329494415203272, 0.10334431946318551, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599, 0.09341208289772931, 0.20329494415203272, 0.10334431946318551, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.09341208289772934, 0.20329494415203272, 0.10334431946318554, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599, 0.09341208289772934, 0.20329494415203272, 0.10334431946318554, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599 ], "expected": [ 0.09341208289772931, 0.20329494415203272, 0.10334431946318551, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599, 0.09341208289772931, 0.20329494415203272, 0.10334431946318551, 0.19832876335348953, 0.12320879259409798, 0.1883964017564031, 0.0834798463322731, 0.20826112495057594, 0.11327655602864176, 0.19336258255494632, 0.13314102915955417, 0.1834302209578599 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a256.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708, -0.8783085599831533, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831533, 0.8783096892353224, -1.4638475999719223, -1.4638494820588708, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588708 ], "expected": [ -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706, -0.8783085599831534, -0.8783096892353224, -0.29276951999438444, -0.29276989641177414, 0.8783085599831534, 0.8783096892353224, -1.4638475999719223, -1.4638494820588706, 0.29276951999438444, 0.29276989641177414, 1.4638475999719223, 1.4638494820588706 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26691026.329545453, 106764102.69318181 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536742, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536742, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "expected": [ -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905, -0.8783100656536574, -0.8783100656536743, -0.2927700218845525, -0.2927700218845581, 0.8783100656536574, 0.8783100656536743, -1.4638501094227623, -1.4638501094227905, 0.2927700218845525, 0.2927700218845581, 1.4638501094227623, 1.4638501094227905 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.329545453, 106764102.69318181 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23354648.063210227, 93418589.9559659 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723, -0.8782859758544261, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544261, 0.8783040430180339, -1.463809959757377, -1.463840071696723, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.463840071696723 ], "expected": [ -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233, -0.8782859758544262, -0.8783040430180339, -0.2927619919514754, -0.29276801433934463, 0.8782859758544262, 0.8783040430180339, -1.463809959757377, -1.4638400716967233, 0.2927619919514754, 0.29276801433934463, 1.463809959757377, 1.4638400716967233 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.063210227, 93418589.9559659 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836, -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836 ], "expected": [ -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836, -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836, -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836 ], "expected": [ -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836, -0.09306783178635186, -0.09306783293049227, -0.09286090660885224, -0.09286090775044879, -0.09244705625385301, -0.09244705739036184, -0.09327475696385149, -0.09327475811053575, -0.09265398143135263, -0.09265398257040532, -0.09224013107635338, -0.09224013221031836 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineFalse/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "before", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/output", "mode": "after", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "before", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/output", "mode": "after", "passed": true, "actual": [ -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531, -1.06746283997473, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.56746283997473, -0.5337322669264918, -1.9457713999578834, 1.222887111544153, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441531 ], "expected": [ -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529, -1.0674628399747301, 0.7837322669264918, -0.18915427999157663, 0.3445774223088306, 1.5674628399747301, -0.5337322669264918, -1.9457713999578834, 1.2228871115441529, 0.6891542799915766, -0.0945774223088306, 2.4457713999578834, -0.9728871115441529 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackFalse/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "before", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "before", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "before", "passed": true, "actual": [ 26691026.329545453, 106764102.69318181 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/output", "mode": "after", "passed": true, "actual": [ -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402556, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402556, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "expected": [ -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928, -1.067465098480486, 0.7837325492402557, -0.18915503282682872, 0.34457751641341855, 1.567465098480486, -0.5337325492402557, -1.9457751641341434, 1.2228875820670928, 0.6891550328268288, -0.09457751641341855, 2.4457751641341434, -0.9728875820670928 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_mean", "mode": "after", "passed": true, "actual": [ 512.0, 1024.0 ], "expected": [ 512.0, 1024.0 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/running_var", "mode": "after", "passed": true, "actual": [ 26691026.329545453, 106764102.69318181 ], "expected": [ 26691026.329545453, 106764102.69318181 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train0/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "before", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "before", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "before", "passed": true, "actual": [ 23354648.063210227, 93418589.9559659 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/output", "mode": "after", "passed": true, "actual": [ -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816392, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816392, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "expected": [ -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424, -1.0674289637816394, 0.7837280322635254, -0.18914298792721307, 0.3445760107545085, 1.5674289637816394, -0.5337280322635254, -1.9457149396360656, 1.2228800537725424, 0.6891429879272131, -0.09457601075450847, 2.4457149396360656, -0.9728800537725424 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/shape", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/input_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_mean", "mode": "after", "passed": true, "actual": [ 448.265625, 896.53125 ], "expected": [ 448.265625, 896.53125 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/running_var", "mode": "after", "passed": true, "actual": [ 23354648.063210227, 93418589.9559659 ], "expected": [ 23354648.063210227, 93418589.9559659 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/train1/state_dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "before", "passed": true, "actual": [ 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876, 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876 ], "expected": [ 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876, 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/output", "mode": "after", "passed": true, "actual": [ 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876, 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876 ], "expected": [ 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876, 0.1103982523204722, 0.19480087469786922, 0.11070864008672165, 0.1946456808128366, 0.1113294156192205, 0.19433529304277136, 0.11008786455422276, 0.19495606858290182, 0.11101902785297105, 0.194490486927804, 0.11163980338546992, 0.19418009915773876 ], "rtol": 1e-11, "atol": 1e-14 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/dtype", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "mlx.core.float64/(2, 2, 3, 2)/a8192.0/affineTrue/trackTrue/eval/state_unchanged", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a1.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3369140625, -0.7330322265625 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a1.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a1.0/input", "mode": "before", "passed": true, "actual": [ -1.3740234375, -0.247314453125, 2.0625, 0.37060546875, -0.6884765625, -0.12353515625 ], "expected": [ -1.3740234375, -0.2469482421875, 2.0625, 0.370361328125, -0.68701171875, -0.12347412109375 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a1.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a1.0/state_mean", "mode": "before", "passed": true, "actual": [ 0.041656494140625, 0.08331298828125 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a1.0/state_var", "mode": "before", "passed": true, "actual": [ 1.166748046875, 2.0419921875 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a1.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73590087890625 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a1.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a1.0/input", "mode": "after", "passed": true, "actual": [ -1.375, -0.247314453125, 2.0625, 0.37060546875, -0.6875, -0.1236572265625 ], "expected": [ -1.3740234375, -0.2469482421875, 2.0625, 0.370361328125, -0.68701171875, -0.12347412109375 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a1.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a1.0/state_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a1.0/state_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a256.0/gamma", "mode": "before", "passed": false, "actual": [ 0.0, 0.0 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a256.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a256.0/input", "mode": "before", "passed": false, "actual": [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.005367279052734375, -0.0009646415710449219, 0.008056640625, 0.0014467239379882812, -0.0026836395263671875, -0.00048232078552246094 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a256.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a256.0/state_mean", "mode": "before", "passed": true, "actual": [ 10.6640625, 21.328125 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a256.0/state_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a256.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73590087890625 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a256.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a256.0/input", "mode": "after", "passed": true, "actual": [ -0.00537109375, -0.0009660720825195312, 0.008056640625, 0.0014476776123046875, -0.002685546875, -0.0004830360412597656 ], "expected": [ -0.005367279052734375, -0.0009646415710449219, 0.008056640625, 0.0014467239379882812, -0.0026836395263671875, -0.00048232078552246094 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a256.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a256.0/state_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a256.0/state_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a4096.0/gamma", "mode": "before", "passed": false, "actual": [ 0.0, 0.0 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a4096.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a4096.0/input", "mode": "before", "passed": false, "actual": [ 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 ], "expected": [ -0.00033545494079589844, -6.031990051269531e-05, 0.0005035400390625, 9.042024612426758e-05, -0.00016772747039794922, -3.0159950256347656e-05 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a4096.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a4096.0/state_mean", "mode": "before", "passed": true, "actual": [ 170.625, 341.25 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a4096.0/state_var", "mode": "before", "passed": false, "actual": [ "Infinity", "Infinity" ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a4096.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73590087890625 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a4096.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.012, "atol": 1.2e-05 }, { "name": "gradient/mlx.core.float16/a4096.0/input", "mode": "after", "passed": true, "actual": [ -0.000335693359375, -6.03795051574707e-05, 0.0005035400390625, 9.047985076904297e-05, -0.0001678466796875, -3.0159950256347656e-05 ], "expected": [ -0.00033545494079589844, -6.031990051269531e-05, 0.0005035400390625, 9.042024612426758e-05, -0.00016772747039794922, -3.0159950256347656e-05 ], "rtol": 0.012, "atol": 1.1920928955078125e-07 }, { "name": "gradient/mlx.core.float16/a4096.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float16/a4096.0/state_mean", "mode": "after", "passed": true, "actual": [ 170.6666717529297, 341.3333435058594 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.float16/a4096.0/state_var", "mode": "after", "passed": true, "actual": [ 4893356.0, 19573420.0 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 0.004, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a1.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a1.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a1.0/input", "mode": "before", "passed": true, "actual": [ -1.375, -0.24609375, 2.0625, 0.37109375, -0.6875, -0.12109375 ], "expected": [ -1.375, -0.2470703125, 2.0625, 0.37109375, -0.6875, -0.12353515625 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a1.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a1.0/state_mean", "mode": "before", "passed": true, "actual": [ 0.041748046875, 0.08349609375 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a1.0/state_var", "mode": "before", "passed": true, "actual": [ 1.166015625, 2.0390625 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a1.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a1.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a1.0/input", "mode": "after", "passed": true, "actual": [ -1.375, -0.248046875, 2.0625, 0.37109375, -0.6875, -0.1240234375 ], "expected": [ -1.375, -0.2470703125, 2.0625, 0.37109375, -0.6875, -0.12353515625 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a1.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a1.0/state_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a1.0/state_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a256.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a256.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a256.0/input", "mode": "before", "passed": true, "actual": [ -0.00537109375, -0.0009613037109375, 0.008056640625, 0.0014495849609375, -0.002685546875, -0.0004730224609375 ], "expected": [ -0.00537109375, -0.000965118408203125, 0.008056640625, 0.0014495849609375, -0.002685546875, -0.0004825592041015625 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a256.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a256.0/state_mean", "mode": "before", "passed": true, "actual": [ 10.6875, 21.375 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a256.0/state_var", "mode": "before", "passed": true, "actual": [ 19072.875, 76288.875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a256.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a256.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a256.0/input", "mode": "after", "passed": true, "actual": [ -0.00537109375, -0.00096893310546875, 0.008056640625, 0.0014495849609375, -0.002685546875, -0.000484466552734375 ], "expected": [ -0.00537109375, -0.000965118408203125, 0.008056640625, 0.0014495849609375, -0.002685546875, -0.0004825592041015625 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a256.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a256.0/state_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a256.0/state_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/input", "mode": "before", "passed": true, "actual": [ -0.000335693359375, -6.008148193359375e-05, 0.0005035400390625, 9.059906005859375e-05, -0.0001678466796875, -2.956390380859375e-05 ], "expected": [ -0.000335693359375, -6.031990051269531e-05, 0.0005035400390625, 9.059906005859375e-05, -0.0001678466796875, -3.0159950256347656e-05 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/state_mean", "mode": "before", "passed": true, "actual": [ 171.0, 342.0 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/state_var", "mode": "before", "passed": true, "actual": [ 4882433.0, 19529728.0 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3359375, -0.73779296875 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 0.07500000000000001, "atol": 7.500000000000001e-05 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/input", "mode": "after", "passed": true, "actual": [ -0.000335693359375, -6.0558319091796875e-05, 0.0005035400390625, 9.059906005859375e-05, -0.0001678466796875, -3.0279159545898438e-05 ], "expected": [ -0.000335693359375, -6.031990051269531e-05, 0.0005035400390625, 9.059906005859375e-05, -0.0001678466796875, -3.0159950256347656e-05 ], "rtol": 0.07500000000000001, "atol": 1e-11 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/state_mean", "mode": "after", "passed": true, "actual": [ 170.6666717529297, 341.3333435058594 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.bfloat16/a4096.0/state_var", "mode": "after", "passed": true, "actual": [ 4893356.0, 19573420.0 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 0.025, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a1.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3363016843795776, -0.7349677085876465 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a1.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a1.0/input", "mode": "before", "passed": true, "actual": [ -1.3744779825210571, -0.24697792530059814, 2.0617237091064453, 0.3704667091369629, -0.6872454881668091, -0.12348872423171997 ], "expected": [ -1.3744783401489258, -0.24697795510292053, 2.0617239475250244, 0.3704667091369629, -0.6872456073760986, -0.12348875403404236 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a1.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a1.0/state_mean", "mode": "before", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a1.0/state_var", "mode": "before", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a1.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3363016843795776, -0.7349677085876465 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a1.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a1.0/input", "mode": "after", "passed": true, "actual": [ -1.3744779825210571, -0.24697792530059814, 2.0617237091064453, 0.3704667091369629, -0.6872454881668091, -0.12348872423171997 ], "expected": [ -1.3744783401489258, -0.24697795510292053, 2.0617239475250244, 0.3704667091369629, -0.6872456073760986, -0.12348875403404236 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a1.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a1.0/state_mean", "mode": "after", "passed": true, "actual": [ 0.0416666679084301, 0.0833333358168602 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a1.0/state_var", "mode": "after", "passed": true, "actual": [ 1.1666667461395264, 2.0416667461395264 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a256.0/gamma", "mode": "before", "passed": true, "actual": [ -1.336306095123291, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a256.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a256.0/input", "mode": "before", "passed": true, "actual": [ -0.005369086749851704, -0.000964757869951427, 0.008053631521761417, 0.0014471368631348014, -0.0026845443062484264, -0.0004823788767680526 ], "expected": [ -0.005369087681174278, -0.0009647579281590879, 0.008053631521761417, 0.0014471368631348014, -0.002684543840587139, -0.00048237896407954395 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a256.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a256.0/state_mean", "mode": "before", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a256.0/state_var", "mode": "before", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a256.0/gamma", "mode": "after", "passed": true, "actual": [ -1.336306095123291, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a256.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a256.0/input", "mode": "after", "passed": true, "actual": [ -0.005369086749851704, -0.000964757869951427, 0.008053631521761417, 0.0014471368631348014, -0.0026845443062484264, -0.0004823788767680526 ], "expected": [ -0.005369087681174278, -0.0009647579281590879, 0.008053631521761417, 0.0014471368631348014, -0.002684543840587139, -0.00048237896407954395 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a256.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a256.0/state_mean", "mode": "after", "passed": true, "actual": [ 10.666666984558105, 21.33333396911621 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a256.0/state_var", "mode": "after", "passed": true, "actual": [ 19115.54296875, 76459.546875 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a4096.0/gamma", "mode": "before", "passed": true, "actual": [ -1.336306095123291, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a4096.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a4096.0/input", "mode": "before", "passed": true, "actual": [ -0.0003355679218657315, -6.0297366871964186e-05, 0.0005033519701100886, 9.044605394592509e-05, -0.00016778401914052665, -3.0148679798003286e-05 ], "expected": [ -0.0003355679800733924, -6.0297370509942994e-05, 0.0005033519701100886, 9.044605394592509e-05, -0.0001677839900366962, -3.0148685254971497e-05 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a4096.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a4096.0/state_mean", "mode": "before", "passed": true, "actual": [ 170.6666717529297, 341.3333435058594 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a4096.0/state_var", "mode": "before", "passed": true, "actual": [ 4893356.0, 19573420.0 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a4096.0/gamma", "mode": "after", "passed": true, "actual": [ -1.336306095123291, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a4096.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 1.8e-05, "atol": 1.8000000000000002e-08 }, { "name": "gradient/mlx.core.float32/a4096.0/input", "mode": "after", "passed": true, "actual": [ -0.0003355679218657315, -6.0297366871964186e-05, 0.0005033519701100886, 9.044605394592509e-05, -0.00016778401914052665, -3.0148679798003286e-05 ], "expected": [ -0.0003355679800733924, -6.0297370509942994e-05, 0.0005033519701100886, 9.044605394592509e-05, -0.0001677839900366962, -3.0148685254971497e-05 ], "rtol": 1.8e-05, "atol": 1e-12 }, { "name": "gradient/mlx.core.float32/a4096.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float32/a4096.0/state_mean", "mode": "after", "passed": true, "actual": [ 170.6666717529297, 341.3333435058594 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float32/a4096.0/state_var", "mode": "after", "passed": true, "actual": [ 4893356.0, 19573420.0 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 6e-06, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a1.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3363019227981567, -0.734967827796936 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a1.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a1.0/input", "mode": "before", "passed": true, "actual": [ -1.3744782873830617, -0.24697795075265516, 2.061723873917404, 0.37046670465465936, -0.6872455865343423, -0.12348875390200426 ], "expected": [ -1.3744782873830619, -0.24697795075265513, 2.061723873917404, 0.37046670465465936, -0.6872455865343422, -0.12348875390200421 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a1.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a1.0/state_mean", "mode": "before", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a1.0/state_var", "mode": "before", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a1.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3363019227981567, -0.734967827796936 ], "expected": [ -1.3363019227981567, -0.734967827796936 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a1.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a1.0/input", "mode": "after", "passed": true, "actual": [ -1.3744782873830617, -0.24697795075265516, 2.061723873917404, 0.37046670465465936, -0.6872455865343423, -0.12348875390200426 ], "expected": [ -1.3744782873830619, -0.24697795075265513, 2.061723873917404, 0.37046670465465936, -0.6872455865343422, -0.12348875390200421 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a1.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a1.0/state_mean", "mode": "after", "passed": true, "actual": [ 0.041666666666666664, 0.08333333333333333 ], "expected": [ 0.041666666666666664, 0.08333333333333333 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a1.0/state_var", "mode": "after", "passed": true, "actual": [ 1.1666666666666667, 2.041666666666667 ], "expected": [ 1.1666666666666667, 2.041666666666667 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a256.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3363062143325806, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a256.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a256.0/input", "mode": "before", "passed": true, "actual": [ -0.005369087448650749, -0.0009647579010118945, 0.008053631173360151, 0.0014471368515046406, -0.002684543724709402, -0.00048237895049274625 ], "expected": [ -0.00536908744865075, -0.0009647579010118944, 0.008053631173360151, 0.0014471368515046406, -0.002684543724709401, -0.00048237895049274625 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a256.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a256.0/state_mean", "mode": "before", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a256.0/state_var", "mode": "before", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a256.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3363062143325806, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a256.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a256.0/input", "mode": "after", "passed": true, "actual": [ -0.005369087448650749, -0.0009647579010118945, 0.008053631173360151, 0.0014471368515046406, -0.002684543724709402, -0.00048237895049274625 ], "expected": [ -0.00536908744865075, -0.0009647579010118944, 0.008053631173360151, 0.0014471368515046406, -0.002684543724709401, -0.00048237895049274625 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a256.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a256.0/state_mean", "mode": "after", "passed": true, "actual": [ 10.666666666666666, 21.333333333333332 ], "expected": [ 10.666666666666666, 21.333333333333332 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a256.0/state_var", "mode": "after", "passed": true, "actual": [ 19115.541666666668, 76459.54166666667 ], "expected": [ 19115.541666666668, 76459.54166666667 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a4096.0/gamma", "mode": "before", "passed": true, "actual": [ -1.3363062143325806, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a4096.0/beta", "mode": "before", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a4096.0/input", "mode": "before", "passed": true, "actual": [ -0.00033556796557072753, -6.0297368813510245e-05, 0.000503351948356185, 9.044605322026212e-05, -0.00016778398278545752, -3.0148684406751904e-05 ], "expected": [ -0.0003355679655707276, -6.029736881351024e-05, 0.0005033519483561851, 9.044605322026212e-05, -0.00016778398278545747, -3.014868440675188e-05 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a4096.0/frozen_state", "mode": "before", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a4096.0/state_mean", "mode": "before", "passed": true, "actual": [ 170.66666666666666, 341.3333333333333 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a4096.0/state_var", "mode": "before", "passed": true, "actual": [ 4893355.541666667, 19573419.541666668 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a4096.0/gamma", "mode": "after", "passed": true, "actual": [ -1.3363062143325806, -0.7349684238433838 ], "expected": [ -1.3363062143325806, -0.7349684238433838 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a4096.0/beta", "mode": "after", "passed": true, "actual": [ 0.5, 2.75 ], "expected": [ 0.5, 2.75 ], "rtol": 3e-11, "atol": 3e-14 }, { "name": "gradient/mlx.core.float64/a4096.0/input", "mode": "after", "passed": true, "actual": [ -0.00033556796557072753, -6.0297368813510245e-05, 0.000503351948356185, 9.044605322026212e-05, -0.00016778398278545752, -3.0148684406751904e-05 ], "expected": [ -0.0003355679655707276, -6.029736881351024e-05, 0.0005033519483561851, 9.044605322026212e-05, -0.00016778398278545747, -3.014868440675188e-05 ], "rtol": 3e-11, "atol": 1e-12 }, { "name": "gradient/mlx.core.float64/a4096.0/frozen_state", "mode": "after", "passed": true, "actual": [ 1 ], "expected": [ 1 ], "rtol": 0, "atol": 0 }, { "name": "gradient/mlx.core.float64/a4096.0/state_mean", "mode": "after", "passed": true, "actual": [ 170.66666666666666, 341.3333333333333 ], "expected": [ 170.66666666666666, 341.3333333333333 ], "rtol": 1e-11, "atol": 1e-06 }, { "name": "gradient/mlx.core.float64/a4096.0/state_var", "mode": "after", "passed": true, "actual": [ 4893355.541666667, 19573419.541666668 ], "expected": [ 4893355.541666667, 19573419.541666668 ], "rtol": 1e-11, "atol": 1e-06 } ] }