Skip to main content
CrystalDocs
Search documentation
Search
Browse Docs
CrystalShards
About
module
ML::NN
Constants
NN_KERNEL_SOURCE = {{ read_file("/tmp/tmp.b
MgOdC
/src/src/ml/nn/../metal/kernels/nn.metal") }}
Load nn.metal kernel source at compile time
Nested types
ML::NN::CrossAttention
ML::NN::Dense
ML::NN::GPUOps
ML::NN::LayerNorm
ML::NN::Linear
ML::NN::MLP
ML::NN::MultiHeadAttention
ML::NN::PatchEmbedding
ML::NN::RMSNorm
ML::NN::TransformerEncoderBlock
ML::NN::ViTEncoder