Skip to main content

Ctrl+K

Installation
Dataset Preparation
Training large language models in xlstm-jax
Configuring Experiments with Hydra
Distributed Training
API Reference
- xlstm_jax
- lmeval_extended_evaluation

.rst

xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice

Contents

Submodules

xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice#

Submodules#

xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice._triton_bw
xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice._triton_fw
xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice.triton_fwbw

previous

xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3.triton_fwbw

next

xlstm_jax.kernels.mlstm_chunkwise.max_triton_fwbw_v3noslice._triton_bw

Contents

Submodules

By NXAI GmbH

© Copyright 2024, NXAI GmbH.