mlx_rs::builder

Trait Builder

Source
pub trait Builder<T: Buildable> {
    type Error: Error;

    // Required method
    fn build(self) -> Result<T, Self::Error>;
}
Expand description

Helper trait for builder

Required Associated Types§

Source

type Error: Error

Error with building

Required Methods§

Source

fn build(self) -> Result<T, Self::Error>

Build the type

Implementors§

Source§

impl Builder<CosineSimilarityLoss> for CosineSimilarityLossBuilder

Source§

impl Builder<GaussianNllLoss> for GaussianNllLossBuilder

Source§

impl Builder<HingeLoss> for HingeLossBuilder

Source§

impl Builder<HuberLoss> for HuberLossBuilder

Source§

impl Builder<KlDivLoss> for KlDivLossBuilder

Source§

impl Builder<L1Loss> for L1LossBuilder

Source§

impl Builder<LogCoshLoss> for LogCoshLossBuilder

Source§

impl Builder<MarginRankingLoss> for MarginRankingLossBuilder

Source§

impl Builder<MseLoss> for MseLossBuilder

Source§

impl Builder<NllLoss> for NllLossBuilder

Source§

impl Builder<SmoothL1Loss> for SmoothL1LossBuilder

Source§

impl Builder<TripletLoss> for TripletLossBuilder

Source§

impl Builder<BatchNorm> for BatchNormBuilder

Source§

impl Builder<Bilinear> for BilinearBuilder

Source§

impl Builder<Celu> for CeluBuilder

Source§

impl Builder<Conv1d> for Conv1dBuilder

Source§

impl Builder<Conv2d> for Conv2dBuilder

Source§

impl Builder<Conv3d> for Conv3dBuilder

Source§

impl Builder<ConvTranspose1d> for ConvTranspose1dBuilder

Source§

impl Builder<ConvTranspose2d> for ConvTranspose2dBuilder

Source§

impl Builder<ConvTranspose3d> for ConvTranspose3dBuilder

Source§

impl Builder<Dropout2d> for Dropout2dBuilder

Source§

impl Builder<Dropout3d> for Dropout3dBuilder

Source§

impl Builder<Dropout> for DropoutBuilder

Source§

impl Builder<Gelu> for GeluBuilder

Source§

impl Builder<Glu> for GluBuilder

Source§

impl Builder<GroupNorm> for GroupNormBuilder

Source§

impl Builder<Gru> for GruBuilder

Source§

impl Builder<InstanceNorm> for InstanceNormBuilder

Source§

impl Builder<LayerNorm> for LayerNormBuilder

Source§

impl Builder<LeakyRelu> for LeakyReluBuilder

Source§

impl Builder<Linear> for LinearBuilder

Source§

impl Builder<LogSoftmax> for LogSoftmaxBuilder

Source§

impl Builder<Lstm> for LstmBuilder

Source§

impl Builder<MultiHeadAttention> for MultiHeadAttentionBuilder

Source§

impl Builder<Prelu> for PreluBuilder

Source§

impl Builder<QuantizedEmbedding> for QuantizedEmbeddingBuilder

Source§

impl Builder<QuantizedLinear> for QuantizedLinearBuilder

Source§

impl Builder<RmsNorm> for RmsNormBuilder

Source§

impl Builder<Rnn> for RnnBuilder

Source§

impl Builder<RotaryPositionalEncoding> for RotaryPositionalEncodingBuilder

Source§

impl Builder<SinusoidalPositionalEncoding> for SinusoidalPositionalEncodingBuilder

Source§

impl Builder<Softmax> for SoftmaxBuilder

Source§

impl Builder<Step> for StepBuilder

Source§

impl Builder<Transformer> for TransformerBuilder

Source§

impl Builder<AdaDelta> for AdaDeltaBuilder

Source§

impl Builder<AdaGrad> for AdaGradBuilder

Source§

impl Builder<Adafactor> for AdafactorBuilder

Source§

impl Builder<Adam> for AdamBuilder

Source§

impl Builder<AdamW> for AdamWBuilder

Source§

impl Builder<Adamax> for AdamaxBuilder

Source§

impl Builder<Lion> for LionBuilder

Source§

impl Builder<RmsProp> for RmsPropBuilder

Source§

impl Builder<Sgd> for SgdBuilder

Source§

impl<'a> Builder<BinaryCrossEntropy<'a>> for BinaryCrossEntropyBuilder<'a>

Source§

impl<'a> Builder<CrossEntropy<'a>> for CrossEntropyBuilder<'a>

Source§

impl<'a> Builder<AlibiInput<'a>> for AlibiInputBuilder<'a>

Source§

impl<'a> Builder<LstmInput<'a>> for LstmInputBuilder<'a>

Source§

impl<'a> Builder<MultiHeadAttentionInput<'a>> for MultiHeadAttentionInputBuilder<'a>

Source§

impl<'a> Builder<RnnInput<'a>> for RnnInputBuilder<'a>

Source§

impl<'a> Builder<RopeInput<'a>> for RopeInputBuilder<'a>