v0.1.4.1

masterfl0w released this 24 Feb 16:26

· 9 commits to main since this release

8aef237

feat:

autogradient implementation for GELU and Dropout layers
mlx usage for GELU and Dropout layers
optimization for CrossEntropyLoss module
fixed pypi version on the website

Assets 2