|
|
c78ce76501
|
Add a simple Module trait and implement it for the various nn layers (#500)
* Start adding the module trait.
* Use the module trait.
* Implement module for qmatmul.
|
2023-08-18 09:38:22 +01:00 |
|
|
|
3eb2bc6d07
|
Softmax numerical stability. (#267)
* Softmax numerical stability.
* Fix the flash-attn test.
|
2023-07-28 13:13:01 +01:00 |
|
|
|
43c7223292
|
Rename the .r functions to .dims so as to be a bit more explicit. (#220)
|
2023-07-22 10:39:27 +01:00 |
|
|
|
b706f32839
|
Add Shape try into (#189)
* Add the TryInto trait for shapes.
* Use the vectorized operations in block mode too.
|
2023-07-18 10:52:16 +01:00 |
|
|
|
d6313d2447
|
Add more tracing details to bert. (#188)
|
2023-07-18 08:11:05 +01:00 |
|
|
|
f0cccd08f0
|
Bert tracing (#184)
* Add some tracing to bert.
* More tracing.
* Add a flag for tracing.
|
2023-07-17 19:40:42 +01:00 |
|
