Laurent Mazare 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

..

encodec_model.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

main.rs

Add some 'cuda-if-available' helper function. (#172)

2023-07-15 08:25:15 +01:00

musicgen_model.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00

nn.rs

Simplify the parameters used by sum and sum_keepdim. (#165)

2023-07-14 08:22:08 +01:00

t5_model.rs

Add a simple Module trait and implement it for the various nn layers (#500)

2023-08-18 09:38:22 +01:00