API reference

The API reference is generated from public Python docstrings by mkdocstrings.

Neural-network modules

standardkernels.nn

Parameter-owning PyTorch modules backed by Standard Kernels.

PreNormGEGLUFFN

Bases: Module

Fixed-shape fused pre-norm GEGLU feed-forward block.

This proof-of-concept module computes x + down(gelu(gate(norm(x))) * up(norm(x))). It currently supports inference on one live module instance, one CUDA stream, and the exact (1024, 768) BF16 input and d_ff=1448 configuration used by the source JUMP kernel.

from_module(module) classmethod

Create an optimized block from a module with the same state dict.

Functional operations

The ops namespace will be populated as stateless public operators are released.