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

UtilitiesΒΆ

  • lrnnx.utils.generation module
    • CUDAGraphStepCache
    • capture_graph()
    • generate()
  • lrnnx.utils.init module
    • make_HiPPO()
    • make_NPLR_HiPPO()
    • make_DPLR_HiPPO()
    • init_log_steps()
    • init_VinvB()
    • init_CV()
Next
lrnnx.utils.generation module
Previous
lrnnx.ops.triton.softplus module
Copyright © 2026, SAiDL
Made with Sphinx and @pradyunsg's Furo