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