Commit Graph

11 Commits

Author SHA1 Message Date
37cad85869 Resurrect the llama npy support. (#140) 2023-07-11 19:32:10 +01:00
64264d97c1 Modular backends (#138)
* Add some trait to formalize backends.

* Use the generic backend trait.
2023-07-11 11:17:02 +01:00
270997a055 Add the elu op. (#113) 2023-07-09 21:56:31 +01:00
dd60bd84bb MKL adjustments. (#87) 2023-07-06 11:37:27 +01:00
a57b314780 Add a batch dimension on the bert example. 2023-07-04 06:10:52 +01:00
ad52b0377c Add the varbuilder + check shapes. 2023-07-03 15:32:20 +01:00
899c76de75 Handle more types in safetensors. 2023-07-03 10:09:46 +01:00
fe2c07e368 Add the ST error. 2023-07-03 08:44:00 +01:00
19cbbc5212 Improve how we check that the dims are in bounds. 2023-06-30 09:11:00 +01:00
b50bd880ce Only narrow when needed + deactivate the kv cache. 2023-06-29 19:07:52 +01:00
d7f729fb8f Refactor the hierarchy. 2023-06-27 11:57:27 +02:00