Add support for automatic mixed precision#31

Open

bpopeters wants to merge 18 commits intomasterfrom

mixed-precision

Commits on Oct 25, 2023

compute exact algo with full precision when using amp
bpopeters
committed

Commits on Oct 26, 2023

Commits on Oct 27, 2023

Commits on Jan 5, 2024

Merge branch 'master' of github.com:deep-spin/entmax into mixed-precision
bpopeters
committed
Merge branch 'mixed-precision' of github.com:deep-spin/entmax into mixed-precision
bpopeters
committed

Commits on May 23, 2024

Merge branch 'master' of github.com:deep-spin/entmax into mixed-precision
bpopeters
committed