Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Hardware
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

jburtoft
/
kda-neuron-kernels

Kernels
kernel
neuron
trainium
kda
linear-attention
fla-core
training
backward
Model card Files Files and versions
xet
Community

Instructions to use jburtoft/kda-neuron-kernels with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Kernels

    How to use jburtoft/kda-neuron-kernels with Kernels:

    # !pip install kernels
    
    from kernels import get_kernel
    
    kernel = get_kernel("jburtoft/kda-neuron-kernels")
  • Notebooks
  • Google Colab
  • Kaggle
kda-neuron-kernels
287 kB
Ctrl+K
Ctrl+K
  • 4 contributors
History: 12 commits
Jim Burtoft
v1.7: add batched recurrent backward (kda_recurrent_bwd_batched); metadata version=3
1bb2101 4 days ago
  • build
    v1.7: add batched recurrent backward (kda_recurrent_bwd_batched); metadata version=3 4 days ago
  • tests
    v1.0: KDA NKI kernels for HF Transformers on AWS Trainium 10 days ago
  • .gitattributes
    1.52 kB
    initial commit 8 days ago
  • .gitignore
    19 Bytes
    v1.1: T-KDA-02 state-decay axis fix + fully model-agnostic 10 days ago
  • LICENSE
    11.3 kB
    v1.0: KDA NKI kernels for HF Transformers on AWS Trainium 10 days ago
  • README.md
    13 kB
    v1.7: add batched recurrent backward (kda_recurrent_bwd_batched); metadata version=3 4 days ago