Daniel Stancl ff066119ca Implement head_mask for Flax BERT and other models copied from BERT (#14620) ... * Implement head_mask for Flax BERT and other models copied from BERT * Remove `from jax._src.nn.functions import sigmoid` Remove `from jax._src.nn.functions import sigmoid` unintentionally added by IDE * Remove no more valid copy statement * Apply patil-suraj's suggestions from code review * Apply suggestions from the code review * Update Flax template * Fix a typo * Also update template for CausalLM modules		2021-12-17 17:06:59 +01:00
..
adding_a_new_example_script	Update namespaces inside torch.utils.data to the latest. (#13167)	2021-08-19 14:29:51 +02:00
adding_a_new_model	Implement head_mask for Flax BERT and other models copied from BERT (#14620)	2021-12-17 17:06:59 +01:00