diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-11 17:04:57 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-11 17:04:57 +0100 |
| commit | 1e26d539d9f9574222e8d049fdbfadfa09e3ce2e (patch) | |
| tree | 0fee8fdca3dd10f47ebbcdb2724249363dbe53aa /candle-pyo3/py_src/candle/nn/container.py | |
| parent | 74497e6bf738366d4c599b42826e204fbcb96f37 (diff) | |
| download | candle-1e26d539d9f9574222e8d049fdbfadfa09e3ce2e.tar.gz candle-1e26d539d9f9574222e8d049fdbfadfa09e3ce2e.tar.bz2 candle-1e26d539d9f9574222e8d049fdbfadfa09e3ce2e.zip | |
Improved mamba model optimized for inference (#1694)
* Sketch the mamba model for inference.
* Complete the forward pass.
* Add the mamba example.
* Optimize the selective-scan part.
* Fix a couple shape mismatches and get inference to work.
* Tweak the readmes.
* More readme tweaks.
Diffstat (limited to 'candle-pyo3/py_src/candle/nn/container.py')
0 files changed, 0 insertions, 0 deletions