Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets
  • 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

Kernels:
jburtoft
/
gte-encoder-neuron-kernels

Kernel card Files Files and versions
xet
Community
gte-encoder-neuron-kernels / tests
39.2 kB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 2 commits
jburtoft's picture
jburtoft
v1.1.0: fused attention block for L = 768 / 1024
fb58c6c 20 days ago
  • README.md
    1.41 kB
    v1.0.0: NKI kernels for NewBERT (gte) encoders on trn2 21 days ago
  • bench_model.py
    3.92 kB
    v1.1.0: fused attention block for L = 768 / 1024 20 days ago
  • conftest.py
    2.91 kB
    v1.0.0: NKI kernels for NewBERT (gte) encoders on trn2 21 days ago
  • test_device.py
    9.47 kB
    v1.1.0: fused attention block for L = 768 / 1024 20 days ago
  • test_device_embed.py
    6.29 kB
    v1.1.0: fused attention block for L = 768 / 1024 20 days ago
  • test_pack_cpu.py
    3.69 kB
    v1.1.0: fused attention block for L = 768 / 1024 20 days ago
  • test_patch_sim.py
    3.66 kB
    v1.0.0: NKI kernels for NewBERT (gte) encoders on trn2 21 days ago
  • test_sim.py
    7.87 kB
    v1.1.0: fused attention block for L = 768 / 1024 20 days ago