• Bartosz Szmelczynski's avatar
    Fix bigbird random attention (#21023) · 88399476
    Bartosz Szmelczynski authored
    * switch np.random.permutation to jax.random.permuation
    
    * remove comments
    
    * remove leftover comment
    
    * skip similarity tests
    
    * modify indices_prng_key usage, add deterministic behaviour
    
    * update style
    
    * remove unused import
    
    * remove copy statement since classes are not identical
    
    * remove numpy import
    
    * revert removing copied from statements
    
    * make style from copied
    
    * remove copied from statement
    
    * update copied from statement to include only np.ndarry
    
    * add deterministic args, unittestskip equivalence tests
    88399476
test_modeling_flax_common.py 58.3 KB