keras-nlp icon indicating copy to clipboard operation
keras-nlp copied to clipboard

Published 20 hours ago • verified publisher icon keras-team

[Tracking] Performance optimizations for library

Open jbischof opened this issue 3 years ago • 1 comments

Discussed in https://github.com/keras-team/keras-nlp/discussions/322

jbischof avatar Aug 29 '22 17:08 jbischof

Opened two more specific issue for this w.r.t. our BERT/RoBERTa models.

  • xla testing for bert - https://github.com/keras-team/keras-nlp/issues/325
  • mixed precision for bert - https://github.com/keras-team/keras-nlp/issues/326

mattdangerw avatar Aug 29 '22 20:08 mattdangerw