Fiix format

google · Nov 7, 2023 · 09f2ec9 · 09f2ec9
1 parent aa9427a
commit 09f2ec9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flax/linen/fp8_ops.py b/flax/linen/fp8_ops.py
@@ -135,6 +135,7 @@ def dot_general_with_precision(
     lhs, rhs, dimension_numbers, precision=lax.Precision.DEFAULT
   )
 
+
 @dot_general_with_precision.defjvp
 def dot_general_with_precision_jvp(
   dimension_numbers, precision, preferred_element_type, primals, tangents