Skip to content

vllm.models.inkling.nvidia.ops

Inkling kernels (NVIDIA).

rmsnorm / sconv import eagerly. The SwiGLU kernels and the FA4 relative-attention wrapper are exposed lazily to keep this package's import path lightweight.

Modules:

  • fa4_rel_attention
  • fa4_warmup

    Startup compilation of the FA4 kernels used by Inkling.

  • lamport

    Deadlock-free fused RS + short-conv + AG + residual + RMSNorm.

  • mm_towers

    Fused CUDA kernels for the Inkling vision/audio towers.

  • norm
  • sconv

    Inkling short-convolution kernels backed by a paged sliding-window state cache.

  • silu_and_mul

    SwiGLU kernels for the Inkling MLP layers.