Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
Synax documentation
Logo
Synax documentation

Contents:

  • Installation
  • Usage
  • Contributing
  • API Reference
    • synax
      • synax.Attention
        • synax.Attention.init
        • synax.Attention.apply
        • synax.Attention.parameter_loss
      • synax.Bias
        • synax.Bias.init
        • synax.Bias.apply
        • synax.Bias.parameter_loss
      • synax.Conv
        • synax.Conv.init
        • synax.Conv.apply
        • synax.Conv.parameter_loss
      • synax.Embed
        • synax.Embed.init
        • synax.Embed.apply
        • synax.Embed.parameter_loss
      • synax.Func
        • synax.Func.init
        • synax.Func.apply
        • synax.Func.parameter_loss
      • synax.Linear
        • synax.Linear.init
        • synax.Linear.apply
        • synax.Linear.parameter_loss
      • synax.Scale
        • synax.Scale.init
        • synax.Scale.apply
        • synax.Scale.parameter_loss
      • synax.Chain
        • synax.Chain.init
        • synax.Chain.parameter_loss
      • synax.Parallel
        • synax.Parallel.init
        • synax.Parallel.parameter_loss
      • synax.Residual
        • synax.Residual.init
        • synax.Residual.parameter_loss
      • synax.GLU
        • synax.GLU.init
      • synax.AutoEncoder
        • synax.AutoEncoder.init
        • synax.AutoEncoder.parameter_loss
      • synax.PReLU
        • synax.PReLU.init
        • synax.PReLU.parameter_loss
      • synax.GRU
        • synax.GRU.init
      • synax.LSTM
        • synax.LSTM.init
      • synax.MGU
        • synax.MGU.init
      • synax.SimpleRNN
        • synax.SimpleRNN.init
        • synax.SimpleRNN.apply
        • synax.SimpleRNN.parameter_loss
      • synax.MLP
      • synax.AlexNet
      • synax.LeNet
      • synax.layer_norm
      • synax.max_pool
      • synax.mean_pool
      • synax.rms_norm
Back to top

DocumentationΒΆ

Synax is a neural network library for JAX.

Contents:

  • Installation
  • Usage
  • Contributing
  • API Reference
Next
Installation
Copyright © 2025, Carlos Martin
Made with Furo