We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cac2f33 commit bff1b65Copy full SHA for bff1b65
dfdx/src/nn/layers/mod.rs
@@ -25,6 +25,7 @@ mod leaky_relu;
25
mod linear;
26
mod ln;
27
mod log_softmax;
28
+#[cfg(feature = "nightly")]
29
mod mamba_minimal;
30
mod matmul;
31
mod multi_head_attention;
@@ -80,6 +81,7 @@ pub use leaky_relu::LeakyReLU;
80
81
pub use linear::{Linear, LinearConfig, LinearConstConfig};
82
pub use ln::Ln;
83
pub use log_softmax::LogSoftmax;
84
85
pub use mamba_minimal::{
86
stateful::{MambaStateCache, MambaStateCacheConfig},
87
MambaBlock, MambaBlockConfig, MambaBlockConstConfig,
0 commit comments