Bhadresh Savani
|
c1e47bf4fe
|
[Flax] Addition of FlaxPegasus (#13420)
* added initial files
* fixes pipeline
* fixes style and quality
* fixes doc issue and positional encoding
* fixes layer norm and test
* fixes quality issue
* fixes code quality
* removed extra layer norm
* added layer norm back in encoder and decoder
* added more code copy quality checks
* update tests
* Apply suggestions from code review
* fix import
* fix test
Co-authored-by: patil-suraj <surajp815@gmail.com>
|
2021-09-14 17:15:19 +02:00 |
|