Commit Graph

9 Commits

Author SHA1 Message Date
Awni Hannun
27c0a8c002
Add llms subdir + update README (#145)
* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format
2023-12-20 10:22:25 -08:00
Awni Hannun
9c7e996ff0
Merge pull request #51 from jbarrow/main
Update BERT to take advantage of bias param in MultiHeadAttention
2023-12-13 15:20:29 -08:00
Joe Barrow
a577abc313 Cleaner masking code 2023-12-09 21:21:24 -05:00
Awni Hannun
5a5decf767 revert accidental change 2023-12-09 14:58:45 -08:00
Awni Hannun
98f4346c81 black format 2023-12-09 14:15:25 -08:00
Joe Barrow
5d4838b02e Updating BERT model to take advantage of bias param in MultiHeadAttention 2023-12-09 12:07:33 -05:00
Joe Barrow
d873e10dfe Updating README for current example, making python>=3.8 compatibile, and fixing code type 2023-12-09 12:01:58 -05:00
Joe Barrow
7320456226 Cleaning implementation for merge 2023-12-09 10:41:15 -05:00
Joe Barrow
4e5b8ceafe BERT implementation 2023-12-08 05:14:11 -05:00