entmax
torch>1.5
