Laurent Mazare 4afa461b34 Sketch the Falcon model. (#93) ...

* Sketch the Falcon model.

* Add more substance to the falcon example.

* Falcon (wip).

* Falcon (wip again).

* Falcon inference.

* Get the weights from the api and properly generate the model.

* Use the proper model.

* Fix the file/revision names.

* Fix bias handling.

* Recompute the rot embeddings.

* Fix the input shape.

* Add the release-with-debug profile.

* Silly bugfix.

* More bugfixes.

* Stricter shape checking in matmul.

2023-07-06 19:01:21 +01:00

..

backprop.rs

Sketch the conv1d op.

2023-07-04 10:52:34 +01:00

conv.rs

Proper conv1d dispatch.

2023-07-04 11:29:28 +01:00

cpu_backend.rs

MKL adjustments. (#87)

2023-07-06 11:37:27 +01:00

cuda_backend.rs

MKL adjustments. (#87)

2023-07-06 11:37:27 +01:00

device.rs

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

display.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

dtype.rs

Add dtype support.

2023-07-02 20:12:26 +01:00

dummy_cuda_backend.rs

Add more of the conv1d op.

2023-07-04 11:15:45 +01:00

error.rs

MKL adjustments. (#87)

2023-07-06 11:37:27 +01:00

layout.rs

Get the cpu backend to compile.

2023-06-28 14:12:38 +01:00

lib.rs

Add a simpler way to specify the dim index for some ops.

2023-07-05 20:22:43 +01:00

npy.rs

Add the kernels.

2023-06-30 10:26:56 +01:00

op.rs

Sketch the conv1d op.

2023-07-04 10:52:34 +01:00

safetensors.rs

Address comments.

2023-07-03 13:52:27 +02:00

shape.rs

Add a simpler way to specify the dim index for some ops.

2023-07-05 20:22:43 +01:00

storage.rs

Add more of the conv1d op.

2023-07-04 11:15:45 +01:00

strided_index.rs

Get the cpu tests to run.

2023-06-28 14:32:02 +01:00

tensor.rs

Sketch the Falcon model. (#93)

2023-07-06 19:01:21 +01:00

utils.rs

PyTorch like display implementation.

2023-06-27 21:16:35 +01:00