elixir-nx / xla

Pre-compiled XLA extension
Apache License 2.0
87 stars 20 forks source link

Bump TensorFlow to 2.8.0 #15

Closed jonatanklosko closed 2 years ago

jonatanklosko commented 2 years ago

Just some tiny changes to the BUILD script, and there is one API change relevant for EXLA (https://github.com/elixir-nx/nx/compare/main...jonatanklosko:jk-tf280). With that, all EXLA tests pass for ma locally (cannot tell about GPU though).