add back regex

remove regex

patch

add an interesting regularization from another paper

add an interesting regularization from another paper

maybe more efficient all gather

maybe more efficient all gather

generalize the distributed all gather to any dimension

generalize the distributed all gather to any dimension

generalize the distributed all gather to any dimension