Fix format

EnzymeAD · Dec 19, 2023 · a0898c5 · a0898c5
1 parent 00e2ea9
commit a0898c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/llama.py b/test/llama.py
@@ -4,6 +4,7 @@
 import jax.lax
 import enzymead.jax as enzyme_jax
 
+
 def rmsnorm(x, weight):
     ss = 1 / jnp.sqrt(x.dot(x) / x.shape[0] + 1e-5)
     return weight * x * ss