masahi/libflash_attn masahi/libflash_attn

  • Stars
    star
    14
  • Rank 1,429,993 (Top 29 %)
  • Language
    C++
  • License
    BSD 3-Clause "New...
  • Created about 1 year ago
  • Updated 8 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
masahi/libflash_attn
github

masahi

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

More Repositories

1

torchscript-to-tvm

Cuda
66
star
2

nnvm-vision-demo

Demos interesting image-in, image-out networks running on both NVIDIA and AMD GPUs, with NNVM
Python
49
star
3

tvm-winograd

Test winograd convolution written in TVM for CUDA and AMDGPU
Python
39
star
4

tvm-cutlass-eval

Python
34
star
5

mxnet-cpp-inference

Test MXNet C++ API for doing inference, given a trained model
C++
6
star
6

cutlass_fpA_intB_gemm

C++
6
star
7

tf2-detection-to-tvm

Python
5
star
8

tensorir-experiment

Python
5
star
9

autotvm-test

Benchmark AutoTVM against cuDNN and MIOpen
Python
3
star
10

depth_estimation

Learning to predict depth from single image
Lua
2
star
11

iwsec21_ntt

Assembly
2
star
12

mos

Multi organ segmentation, undergrad course project
Python
1
star
13

hb_tvm_example

Python
1
star
14

ocaml_practice

OCaml
1
star