• Bhadresh Savani's avatar
    [Flax] Addition of FlaxPegasus (#13420) · c1e47bf4
    Bhadresh Savani authored
    
    
    * 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: default avatarpatil-suraj <surajp815@gmail.com>
    c1e47bf4
index.rst 46.7 KB