Laurent Mazare fa760759e5 Allow for lazy loading of npz files, use it in llama to reduce memory usage in the cpu version. (#141)

2023-07-11 20:22:34 +01:00

..

activation.rs

[nn] Move the Embedding and Activation parts. (#116)

2023-07-10 10:24:52 +01:00

conv.rs

Move the conv1d layer to candle_nn. (#117)

2023-07-10 11:02:06 +01:00

embedding.rs

[nn] Move the Embedding and Activation parts. (#116)

2023-07-10 10:24:52 +01:00

init.rs

Sketch the tensor initialization module. (#134)

2023-07-11 07:41:46 +01:00

layer_norm.rs

Sketch the candle-nn crate. (#115)

2023-07-10 08:50:09 +01:00

lib.rs

Sketch the tensor initialization module. (#134)

2023-07-11 07:41:46 +01:00

linear.rs

Sketch the candle-nn crate. (#115)

2023-07-10 08:50:09 +01:00

var_builder.rs

Allow for lazy loading of npz files, use it in llama to reduce memory usage in the cpu version. (#141)

2023-07-11 20:22:34 +01:00