Commit History
Added NIE on xla device
f5895e7
added cuda as optional
85a3cf8
Feature: Add mixed precision support and direct bfloat16 support.
1940326
daniel shalem
commited on