Skip to main content
Back to top
Ctrl
+
K
ejkernel documentation
Getting Started:
Installation
Architecture & Design:
ejKernel Project Overview
Kernel Registry System Analysis
Ops System Architecture Analysis
MaskInfo Guide: Comprehensive Attention Mask Management
Kernel Implementations Analysis
Module Operations Layer Analysis
Utilities and Helper Functions Analysis
Test Suite and Examples Analysis
ejKernel Comprehensive Architecture Report
User Guides:
ejkernel API Documentation
ejkernel.modules - High-Level Operations API
ejkernel.types - Masking and Type Utilities
ejkernel.ops - Kernel Framework and Configuration Management
API Reference:
ejkernel API Reference
ejkernel.callib package
ejkernel.callib._ejit
ejkernel.callib._pallas_call
ejkernel.callib._triton_call
ejkernel.callib._utils
ejkernel.kernels package
ejkernel.kernels._pallas package
ejkernel.kernels._pallas.gpu package
ejkernel.kernels._pallas.tpu package
ejkernel.kernels._triton package
ejkernel.kernels._triton.blocksparse_attention package
ejkernel.kernels._triton.flash_attention package
ejkernel.kernels._triton.flash_mla package
ejkernel.kernels._triton.gla package
ejkernel.kernels._triton.lightning_attn package
ejkernel.kernels._triton.mean_pooling package
ejkernel.kernels._triton.native_sparse_attention package
ejkernel.kernels._triton.page_attention package
ejkernel.kernels._triton.ragged_decode_attention package
ejkernel.kernels._triton.ragged_page_attention_v2 package
ejkernel.kernels._triton.ragged_page_attention_v3 package
ejkernel.kernels._triton.recurrent package
ejkernel.kernels._triton.ring_attention package
ejkernel.kernels._triton.unified_attention package
ejkernel.kernels._xla package
ejkernel.kernels._xla.attention package
ejkernel.kernels._xla.blocksparse_attention package
ejkernel.kernels._xla.flash_attention package
ejkernel.kernels._xla.gla package
ejkernel.kernels._xla.grouped_matmul package
ejkernel.kernels._xla.kernel_delta_attention package
ejkernel.kernels._xla.lightning_attn package
ejkernel.kernels._xla.mean_pooling package
ejkernel.kernels._xla.native_sparse_attention package
ejkernel.kernels._xla.page_attention package
ejkernel.kernels._xla.prefill_page_attention package
ejkernel.kernels._xla.ragged_decode_attention package
ejkernel.kernels._xla.ragged_page_attention_v2 package
ejkernel.kernels._xla.ragged_page_attention_v3 package
ejkernel.kernels._xla.recurrent package
ejkernel.kernels._xla.ring_attention package
ejkernel.kernels._xla.scaled_dot_product_attention package
ejkernel.kernels._xla.state_space_v1 package
ejkernel.kernels._xla.state_space_v2 package
ejkernel.kernels._xla.unified_attention package
ejkernel.kernels._registry
ejkernel.modules package
ejkernel.modules.operations package
ejkernel.modules.operations.attention
ejkernel.modules.operations.blocksparse_attention
ejkernel.modules.operations.configs
ejkernel.modules.operations.flash_attention
ejkernel.modules.operations.gated_linear_attention
ejkernel.modules.operations.grouped_matmul
ejkernel.modules.operations.kernel_delta_attention
ejkernel.modules.operations.lightning_attention
ejkernel.modules.operations.multi_head_latent_attention
ejkernel.modules.operations.native_sparse_attention
ejkernel.modules.operations.page_attention
ejkernel.modules.operations.pooling
ejkernel.modules.operations.prefill_page_attention
ejkernel.modules.operations.ragged_decode_attention
ejkernel.modules.operations.ragged_page_attention_v2
ejkernel.modules.operations.ragged_page_attention_v3
ejkernel.modules.operations.recurrent
ejkernel.modules.operations.ring_attention
ejkernel.modules.operations.scaled_dot_product_attention
ejkernel.modules.operations.state_space_v1
ejkernel.modules.operations.state_space_v2
ejkernel.modules.operations.unified_attention
ejkernel.modules.base
ejkernel.ops package
ejkernel.ops.config package
ejkernel.ops.config.cache
ejkernel.ops.config.persistent
ejkernel.ops.config.selection
ejkernel.ops.core package
ejkernel.ops.core.kernel
ejkernel.ops.core.types
ejkernel.ops.execution package
ejkernel.ops.execution.batch
ejkernel.ops.execution.executor
ejkernel.ops.execution.offline
ejkernel.ops.execution.profiler
ejkernel.ops.execution.tuning
ejkernel.ops.utils package
ejkernel.ops.utils.datacarrier
ejkernel.ops.utils.fingerprint
ejkernel.ops.utils.meta
ejkernel.ops.utils.serialize
ejkernel.ops.registry
ejkernel.types package
ejkernel.types.mask
ejkernel.xla_utils package
ejkernel.xla_utils.cumsum
ejkernel.xla_utils.shardings
ejkernel.xla_utils.utils
ejkernel.benchmarks
ejkernel.loggings
ejkernel.utils
.rst
.pdf
ejkernel.kernels._xla.flash_attention._xla_impl_fwd
ejkernel.kernels._xla.flash_attention._xla_impl_fwd
#