fix: tolerances

Author: polvalente

test/axon/compiler_test.exs

@@ -5299,7 +5299,7 @@ defmodule CompilerTest do
 
       input = random({1, 1})
 
-      assert_equal(predict_fn.(params, input), expected_predict_fn.(input, k1, b1, k2, b2))
+      assert_all_close(predict_fn.(params, input), expected_predict_fn.(input, k1, b1, k2, b2), atol: 1.0e-7)
     end
 
     test "predicts correctly with multiple blocks in network" do