Commit Graph

20 Commits

Author SHA1 Message Date
e2acbe1e72 Update the wasm example locations in the readme. (#306) 2023-08-02 11:36:43 +01:00
afb5e24a63 Remove map ownership from save. 2023-08-01 17:19:22 +02:00
89d1fd03e5 Adding new surface for savetensors (global load, global save). 2023-08-01 15:00:38 +02:00
310094310b Modifying safetensors export to get simple load and save. 2023-08-01 15:00:38 +02:00
6242a1470e Starting the book. 2023-08-01 14:26:02 +02:00
fb84ead8f7 Add the starcoder example to the readme. (#268)
* Add the starcoder example to the readme.

* Tweak.
2023-07-28 13:26:23 +01:00
6475bfadfe Simplify Tensor::randn. (#255)
* Simplify Tensor::randn.

* Also switch Tensor::rand to use a generic dtype.

* Support sampling for f16.

* Cleanup.
2023-07-27 07:40:36 +01:00
160ba09d30 Polish the llama2 wasm ui. (#232)
* Polish the llama2 wasm ui.

* readme update.
2023-07-24 15:28:27 +01:00
12d6dc018d Support for MQA for llama v2. (#205)
* Support for MQA for llama v2.

* More llama-v2.

* Move the rotary embedding precomputation in the cache.

* Add a v2 flag.

* Use the hf model.
2023-07-20 06:39:04 +01:00
dcb4a9291e Expliciting how to enable cuda. 2023-07-14 17:08:05 +02:00
2d5e952cf9 Updating the example to reflect new command lines. 2023-07-14 16:56:33 +02:00
d88b6cdca9 Add backtrace information to errors where relevant. (#166)
* Add backtrace information to errors where relevant.

* More backtrace information.

* Add to the FAQ.
2023-07-14 09:31:25 +01:00
465fc8c0c5 Add some documentation and test to the linear layer. (#151)
* Add some documentation and test to the linear layer.

* Layer norm doc.

* Minor tweaks.
2023-07-12 20:24:23 +01:00
91817b9e57 Tweak the table formatting. 2023-07-12 18:43:52 +01:00
6e938cfe9d Adding cheatsheet + expand on other ML frameworks. 2023-07-12 18:35:34 +02:00
868743b8b9 Expanding a bit the README 2023-07-10 12:51:37 +02:00
f1e29cd405 Allow using mkl in tests. (#90) 2023-07-06 13:25:05 +01:00
47937650aa And add back some readme :) 2023-06-27 15:50:43 +01:00
d7f729fb8f Refactor the hierarchy. 2023-06-27 11:57:27 +02:00
fe7a5f53a1 Initial commit 2023-06-19 17:06:31 +01:00