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.