mirror of
https://github.com/huggingface/candle.git
synced 2025-06-21 12:20:46 +00:00
1b98f84a2b
* Fast kernels for rotary embeddings. * Add a test for the fast CPU kernel. * Rope cuda bindings. * Cuda kernel. * Metal kernel (part 1). * Cuda kernels. * Finish the metal kernel. * Use the new kernels in the quantized example. * Fix warning.
41 lines
1.2 KiB
Rust
41 lines
1.2 KiB
Rust
pub mod activation;
|
|
pub mod batch_norm;
|
|
pub mod conv;
|
|
pub mod embedding;
|
|
pub mod encoding;
|
|
pub mod func;
|
|
pub mod group_norm;
|
|
pub mod init;
|
|
pub mod layer_norm;
|
|
pub mod linear;
|
|
pub mod loss;
|
|
pub mod ops;
|
|
pub mod optim;
|
|
pub mod rnn;
|
|
pub mod rotary_emb;
|
|
pub mod sequential;
|
|
pub mod var_builder;
|
|
pub mod var_map;
|
|
|
|
pub use activation::{prelu, Activation, PReLU};
|
|
pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
|
|
pub use conv::{
|
|
conv1d, conv1d_no_bias, conv2d, conv2d_no_bias, conv_transpose1d, conv_transpose1d_no_bias,
|
|
conv_transpose2d, conv_transpose2d_no_bias, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig,
|
|
ConvTranspose1d, ConvTranspose1dConfig, ConvTranspose2d, ConvTranspose2dConfig,
|
|
};
|
|
pub use embedding::{embedding, Embedding};
|
|
pub use func::{func, func_t, Func, FuncT};
|
|
pub use group_norm::{group_norm, GroupNorm};
|
|
pub use init::Init;
|
|
pub use layer_norm::{layer_norm, rms_norm, LayerNorm, LayerNormConfig, RmsNorm};
|
|
pub use linear::{linear, linear_b, linear_no_bias, Linear};
|
|
pub use ops::Dropout;
|
|
pub use optim::{AdamW, Optimizer, ParamsAdamW, SGD};
|
|
pub use rnn::{gru, lstm, GRUConfig, LSTMConfig, GRU, LSTM, RNN};
|
|
pub use sequential::{seq, Sequential};
|
|
pub use var_builder::VarBuilder;
|
|
pub use var_map::VarMap;
|
|
|
|
pub use candle::{Module, ModuleT};
|