Alternatives and similar repositories for Needle

Users that are interested in Needle are comparing it to the libraries listed below

• yifanlu0227 / MIT-6.5940
  All Homeworks for TinyML and Efficient Deep Learning Computing 6.5940 • Fall • 2023 • https://efficientml.ai
  ☆190Updated 2 years ago
• harleyszhang / llm_counts
  llm theoretical performance analysis tools and support params, flops, memory and latency analysis.
  ☆115Updated 6 months ago
• mit-han-lab / parallel-computing-tutorial
  ☆177Updated 2 years ago
• xlite-dev / ffpa-attn
  🤖FFPA: Extend FlashAttention-2 with Split-D, ~O(1) SRAM complexity for large headdim, 1.8x~3x↑🎉 vs SDPA EA.
  ☆250Updated this week
• Sunt-ing / stick
  A PyTorch-like deep learning framework. Just for fun.
  ☆157Updated 2 years ago
• 66RING / tiny-flash-attention
  flash attention tutorial written in python, triton, cuda, cutlass
  ☆484Updated 3 weeks ago
• PKUFlyingPig / CMU10-714
  Learning material for CMU10-714: Deep Learning System
  ☆300Updated last year
• nvixnu / pmpp__programming_massively_parallel_processors
  Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…
  ☆77Updated 5 years ago
• eedalong / ECE408
  Code base and slides for ECE408：Applied Parallel Programming On GPU.
  ☆145Updated 4 years ago
• SiriusNEO / Triton-Puzzles-Lite
  Puzzles for learning Triton, play it with minimal environment configuration!
  ☆624Updated last month
• tspeterkim / paged-attention-minimal
  a minimal cache manager for PagedAttention, on top of llama3.
  ☆135Updated last year
• XiaoSongXS / dgemm-knl
  DGEMM on KNL, achieve 75% MKL
  ☆19Updated 3 years ago
• interestingLSY / swiftLLM
  A tiny yet powerful LLM inference system tailored for researching purpose. vLLM-equivalent performance with only 2k lines of code (2% of …
  ☆313Updated 8 months ago
• InternLM / turbomind
  ☆96Updated 10 months ago
• hao-ai-lab / cse234-w25-PA
  ☆46Updated 10 months ago
• Bruce-Lee-LY / decoding_attention
  Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.
  ☆46Updated 7 months ago
• interestingLSY / CUDA-From-Correctness-To-Performance-Code
  Codes & examples for "CUDA - From Correctness to Performance"
  ☆121Updated last year
• openmlsys / openmlsys-cuda
  Tutorials for writing high-performance GPU operators in AI frameworks.
  ☆136Updated 2 years ago
• LDLINGLINGLING / nano_vllm_note
  注释的nano_vllm仓库，并且完成了MiniCPM4的适配以及注册新模型的功能
  ☆158Updated 5 months ago
• CalebDu / Awesome-Cute
  ☆113Updated 8 months ago
• iclementine / optimize_softmax
  Optimize softmax in triton in many cases
  ☆22Updated last year
• ailzhang / EfficientPyTorch
  Code release for book "Efficient Training in PyTorch"
  ☆125Updated 10 months ago
• tile-ai / tilelang-puzzles
  Learning TileLang with 10 puzzles!
  ☆118Updated last week
• xlite-dev / HGEMM
  ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.
  ☆148Updated 9 months ago
• lzyrapx / LeetGPU
  🌈 Solutions of LeetGPU
  ☆70Updated last week
• kilianhae / FlashAttention.C
  Flash Attention in raw Cuda C beating PyTorch
  ☆37Updated last year
• mlc-ai / notebooks
  ☆222Updated last year
• dlsyscourse / hw0
  ☆56Updated 5 months ago
• weishengying / tiny-flash-attention
  使用 cutlass 实现 flash-attention 精简版，具有教学意义
  ☆56Updated last year
• sonnyli / flash_attention_from_scratch
  Flash Attention from Scratch on CUDA Ampere
  ☆129Updated 5 months ago