Laurent Mazare 53510ce427 Use a silu activation in mistral. (#991)

2023-09-29 07:06:54 +01:00

..

activation.rs

Use a silu activation in mistral. (#991)

2023-09-29 07:06:54 +01:00

batch_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

conv.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

embedding.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

func.rs

Add a couple functions required for yolo. (#527)

2023-08-20 17:02:05 +01:00

group_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

init.rs

VarBuilder cleanup (#627)

2023-08-27 18:03:26 +01:00

layer_norm.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

lib.rs

Expose the conv2d-transpose layers. (#761)

2023-09-07 06:04:52 +01:00

linear.rs

Add clone to various nn layers. (#910)

2023-09-20 11:33:51 +01:00

loss.rs

Fix typo in the nll function document (#742)

2023-09-05 09:25:11 +01:00

ops.rs

Fix the leaky relu. (#898)

2023-09-19 18:17:17 +01:00

optim.rs

Add the optimizer trait. (#702)

2023-09-01 12:55:39 +01:00

rnn.rs

Configurable layer idx for the lstm layer. (#962)

2023-09-25 21:31:14 +01:00

var_builder.rs

Depreate the VarBuilder::from_safetensors function. (#951)

2023-09-24 11:18:17 +01:00

var_map.rs

Use yoke to provide a self-referential container for mmaped safetenso… (#939)

2023-09-23 15:43:11 +01:00