summaryrefslogtreecommitdiff
path: root/candle-nn
Commit message (Expand)AuthorAgeFilesLines
* Add a convenient way to rename tensors accessed through a varbuilder. (#2052)Laurent Mazare2024-04-131-0/+93
* Add the rope THD kernel. (#2014)Laurent Mazare2024-04-052-0/+262
* Relax the contiguous check for cuda kernels. (#2000)Laurent Mazare2024-04-031-1/+1
* Add benchmarks for the candle-nn package (#1995)Thomas Santerre2024-04-035-0/+175
* Add fn 'get_with_hints_dtype' in VarBuilder (#1877) (#1897)yinqiwen2024-04-011-4/+15
* Fix detail in new RoPE implementation (#1935)Hugo Abonizio2024-03-251-1/+1
* Contiguous variant of the rope kernel. (#1929)Laurent Mazare2024-03-252-2/+282
* Fast kernels for rotary embeddings. (#1928)Laurent Mazare2024-03-244-0/+277
* RmsNorm kernel for metal. (#1895)Laurent Mazare2024-03-211-1/+46
* Custom op for RmsNorm (#1890)Laurent Mazare2024-03-212-8/+197
* Optimize the cat operation on contiguous tensors (#1855)Laurent Mazare2024-03-171-0/+19
* add clone to candle dropout (#1814)Kirpal Grewal2024-03-081-1/+1
* Improve metal buffer usage (#1807)ivarflakstad2024-03-071-1/+2
* Add the StarCoder2 model. (#1779)Laurent Mazare2024-02-281-0/+4
* Encodec model. (#1771)Laurent Mazare2024-02-271-1/+1
* Tweak the VarMap set type. (#1758)Laurent Mazare2024-02-252-2/+39
* Support for attention bias in gemma + refactor things a bit. (#1744)Laurent Mazare2024-02-222-6/+19
* Bugfix for applying the bias in conv1d-transpose. (#1732)Laurent Mazare2024-02-181-1/+1
* Support for groups in conv-transpose1d. (#1731)Laurent Mazare2024-02-181-3/+13
* Expose the weights and biases in transposed convolutions. (#1727)Laurent Mazare2024-02-181-0/+16
* Expose more conv1d functions/structs. (#1726)Laurent Mazare2024-02-172-2/+19
* feat: add silu activation function (#1706)OlivierDehaene2024-02-142-4/+3
* Detach the tensors on batch-norm eval. (#1702)Laurent Mazare2024-02-131-2/+12
* Fix clippy lints for 1.76. (#1682)Laurent Mazare2024-02-081-1/+1
* Enhance pickle to retrieve state_dict with a given key (#1671)Dilshod Tadjibaev2024-02-061-1/+1
* Add `VarBuilder::from_backend` (#1670)Daniël de Kok2024-02-061-8/+17
* Quantized GGUF style (#1523)Nicolas Patry2024-01-171-1/+4
* Update the Phi model to use the updated architecture. (#1580)Laurent Mazare2024-01-131-0/+1
* Simplifying our internal cargo dependencies. (#1529)Nicolas Patry2024-01-071-2/+2
* Simplify the one-hot implementation, support arbitrary rank. (#1514)Laurent Mazare2024-01-011-181/+38
* Add one-hot/cold encoding (#1489)Ryan Tate2024-01-013-0/+414
* Do not implement Module for BatchNorm. (#1513)Laurent Mazare2024-01-012-15/+15
* Small tweaks to batch-norm. (#1505)Laurent Mazare2023-12-301-19/+16
* [Breaking] Add training to batchnorm with exponential moving average (#1504)nkoppel2023-12-302-50/+169
* Bump the crate version to 0.3.3. (#1490)Laurent Mazare2023-12-281-1/+1
* Merge pull request #1318 from huggingface/metal4Nicolas Patry2023-12-202-0/+44
|\
| * Clippy pass.Nicolas Patry2023-12-181-3/+3
| * Addressing a lot of comments.Nicolas Patry2023-12-151-1/+2
| * Remove `unwrap()`.Nicolas Patry2023-12-151-2/+2
| * Renamed all kernel names.Nicolas Patry2023-12-151-3/+3
| * Fixing softmax.Nicolas Patry2023-12-151-1/+1
| * Working with merging encoders and using fences.Nicolas Patry2023-12-141-2/+0
| * Lots of updates including some stack of command buffers.nicolas2023-12-122-2/+5
| * Starting to fix some tests.Nicolas Patry2023-11-302-0/+42
* | Bump the crate version to 0.3.2. (#1452)Laurent Mazare2023-12-171-1/+1
* | Fix a couple typos (#1451)Laurent Mazare2023-12-171-1/+1
* | Expose AdamW parameters (#1449)Dave Lage2023-12-161-0/+8
* | Speedup ShardedSafeTensors to load Tensors with default hints (#1384)YiiSh2023-12-141-1/+7
* | Another prelu bugfix. (#1407)Laurent Mazare2023-12-061-1/+1
* | Use the proper broadcasting for prelu. (#1406)Laurent Mazare2023-12-051-5/+16
generated by cgit v1.2.3 (git 2.39.1) at 2025-08-05 03:43:59 +0000