Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
lrnnx 1.0.0.post1 documentation
lrnnx 1.0.0.post1 documentation

API Reference

  • Models
    • lrnnx.models.lti package
      • lrnnx.models.lti.base module
      • lrnnx.models.lti.centaurus module
      • lrnnx.models.lti.lru module
      • lrnnx.models.lti.s4 module
      • lrnnx.models.lti.s4d module
      • lrnnx.models.lti.s5 module
    • lrnnx.models.ltv package
      • lrnnx.models.ltv.base module
      • lrnnx.models.ltv.mamba module
      • lrnnx.models.ltv.rglru module
      • lrnnx.models.ltv.s5 module
      • lrnnx.models.ltv.s7 module
  • Architectures
    • lrnnx.architectures.classifier module
    • lrnnx.architectures.embedding module
    • lrnnx.architectures.language_model module
    • lrnnx.architectures.lru_unet module
  • Core
    • lrnnx.core.base module
    • lrnnx.core.convolution module
    • lrnnx.core.discretization module
  • Operations
    • lrnnx.ops.rglru_scan module
    • lrnnx.ops.s4_kernel_interface module
    • lrnnx.ops.s4_utils module
    • lrnnx.ops.s7_scan module
    • lrnnx.ops.selective_scan module
    • lrnnx.ops.simplified_scan module
    • lrnnx.ops.torch module
    • lrnnx.ops.triton package
      • lrnnx.ops.triton.layer_norm module
      • lrnnx.ops.triton.selective_state_update module
      • lrnnx.ops.triton.simplified_state_update module
      • lrnnx.ops.triton.softplus module
  • Utilities
    • lrnnx.utils.generation module
    • lrnnx.utils.init module
Back to top
View this page
Edit this page

ModelsΒΆ

Linear RNN model implementations.

  • lrnnx.models.lti package
    • LTI_LRNN
    • Centaurus
    • CentaurusDWS
    • CentaurusFull
    • CentaurusNeck
    • CentaurusPWNeck
    • LRU
    • S4
    • S4D
    • S5
    • Submodules
  • lrnnx.models.ltv package
    • LTV_LRNN
    • Mamba
    • RGLRU
    • S7
    • S5
    • Submodules
Next
lrnnx.models.lti package
Previous
Home
Copyright © 2026, SAiDL
Made with Sphinx and @pradyunsg's Furo