Hm, when I run the example, I get this error:
> expected scalar type BFloat16 but found Float
Has anyone seen this error? It's pretty hard to google for.
> expected scalar type BFloat16 but found Float
Has anyone seen this error? It's pretty hard to google for.