diff --git a/jax/experimental/mosaic/gpu/examples/BUILD b/jax/experimental/mosaic/gpu/examples/BUILD index fe1a7e9180ac..33b6d015c609 100644 --- a/jax/experimental/mosaic/gpu/examples/BUILD +++ b/jax/experimental/mosaic/gpu/examples/BUILD @@ -48,6 +48,8 @@ py_library( ], ) +# brain +# learning/brain/ jax_multiplatform_test( name = "run_matmul", srcs = ["matmul.py"],