Lighter

A Keras like abstraction layer on top of the Rust ML framework candle

MPL-2.0 License

Downloads
2.3K
Stars
20

Rust Lighter

This project was started as my RUST exercise to abstract the Rust minimalist ML framework Candle (https://github.com/huggingface/candle) and introduce a more convenient way of programming neural network machine learning models.

The behaviour is inspired by Python KERAS (https://keras.io) and the initial step based on the Rust-Keras-like code (https://github.com/AhmedBoin/Rust-Keras-Like).

So let's call the project Candle Lighter πŸ•―, because it helps to turn on the candle light and is even easier to implement.

Examples can be found below the lib/examples/ directory.

To use it as library just call 'cargo add candlelighter'

MAINTAINERS AND CONTRIBUTORS ARE HIGHLY WELCOME

Note: It is by far not production ready and is only used for own training purposes. No warranty and liability is given. I am a private person and not targeting any commercial benefits.

Supported Layer types

Meta Layer Type State Example
Sequential model - βœ…
- Feature scaling πŸƒ DNN and TNN
- Dense βœ… DNN
- Convolution βœ… CNN
- Pooling βœ… -
- Normalization βœ… -
- Flatten βœ… -
- Recurrent βœ… RNN 1st throw
- Regulation βœ… -
- Recurrent βœ… RNN 1st throw
- Autoencoder πŸƒ -
- Feature embedding βœ… S2S 1st throw
- Attention πŸƒ TNN 1st throw
- Mixture of Experts πŸƒ ENN 1st throw
- Feature masking and -quantization πŸƒ -
- KAN-Dense πŸƒ -
Model fine tuning (PEFT) - πŸƒ In development: DNN2 & DNN3
Parallel model (in sense of split) - πŸƒ PNN 1st throw
Parallel model Merging πŸƒ PNN 1st throw
Transformer models see πŸƒ
* BERT Text similarity βœ… LLM
* LLAMA Completion (Chat) βœ… LLM2
Reinforcement models see πŸƒ

License

Tripple-licensed to be compatible with the Rust project and the source roots.

Licensed under the MPL 2.0, MIT license or the Apache license, Version 2.0 at your option.

Package Rankings
Related Projects