Hide navigation sidebar

Hide table of contents sidebar

Skip to content

Toggle site navigation sidebar

Synax documentation

Toggle table of contents sidebar

Synax documentation

Contents:

Installation
Usage
Contributing
API Reference
Toggle navigation of API Reference
- synax
  Toggle navigation of synax
  - synax.Attention
    Toggle navigation of synax.Attention
  - synax.Bias
    Toggle navigation of synax.Bias
  - synax.Conv
    Toggle navigation of synax.Conv
  - synax.Embed
    Toggle navigation of synax.Embed
  - synax.Func
    Toggle navigation of synax.Func
  - synax.Linear
    Toggle navigation of synax.Linear
  - synax.Scale
    Toggle navigation of synax.Scale
  - synax.Chain
    Toggle navigation of synax.Chain
    - synax.Chain.init_params
    - synax.Chain.param_loss
  - synax.Parallel
    Toggle navigation of synax.Parallel
    - synax.Parallel.init_params
    - synax.Parallel.param_loss
  - synax.Residual
    Toggle navigation of synax.Residual
    - synax.Residual.init_params
    - synax.Residual.param_loss
  - synax.GLU
    Toggle navigation of synax.GLU
    - synax.GLU.init_params
    - synax.GLU.param_loss
  - synax.AutoEncoder
    Toggle navigation of synax.AutoEncoder
    - synax.AutoEncoder.init_params
    - synax.AutoEncoder.param_loss
  - synax.PReLU
    Toggle navigation of synax.PReLU
    - synax.PReLU.param_loss
  - synax.GRU
    Toggle navigation of synax.GRU
    - synax.GRU.init_params
    - synax.GRU.init_state
  - synax.LSTM
    Toggle navigation of synax.LSTM
    - synax.LSTM.init_params
    - synax.LSTM.init_state
  - synax.MGU
    Toggle navigation of synax.MGU
    - synax.MGU.init_params
    - synax.MGU.init_state
  - synax.SimpleRNN
    Toggle navigation of synax.SimpleRNN
  - synax.AntiSymmetricMatrix
    Toggle navigation of synax.AntiSymmetricMatrix
    - synax.AntiSymmetricMatrix.init_params
    - synax.AntiSymmetricMatrix.param_loss
  - synax.SpecialOrthogonalMatrix
    Toggle navigation of synax.SpecialOrthogonalMatrix
    - synax.SpecialOrthogonalMatrix.init_params
  - synax.SymmetricMatrix
    Toggle navigation of synax.SymmetricMatrix
    - synax.SymmetricMatrix.init_params
    - synax.SymmetricMatrix.param_loss
  - synax.Constant
    Toggle navigation of synax.Constant
    - synax.Constant.init_params
    - synax.Constant.param_loss
  - synax.MLP
  - synax.AlexNet
  - synax.LeNet
  - synax.layer_norm
  - synax.max_pool
  - synax.mean_pool
  - synax.rms_norm

Toggle table of contents sidebar

synax.Scale.param_loss¶

synax.Scale.param_loss(params: jax.Array) → jax.Array | float[source]¶

Parameter loss.

Parameters:: params – Parameters.
Returns:: Scalar.

synax.Scale.apply

Copyright © 2025, Carlos Martin

Made with Furo

On this page

synax.Scale.param_loss
- synax.Scale.param_loss()