LaurentMazare/jax-flash-attn2 LaurentMazare/jax-flash-attn2

  • Stars
    star
    5
  • Rank 2,861,937 (Top 57 %)
  • Language
    C++
  • Created 4 months ago
  • Updated 4 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
LaurentMazare/jax-flash-attn2
github

LaurentMazare

Reviews

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

Repository Details

JAX bindings for the flash-attention2 kernels

More Repositories

1

tch-rs

Rust bindings for the C++ api of PyTorch.
Rust
4,162
star
2

diffusers-rs

An implementation of the diffusers api in Rust
Rust
521
star
3

ocaml-torch

OCaml bindings for PyTorch
OCaml
412
star
4

tensorflow-ocaml

OCaml bindings for TensorFlow
OCaml
283
star
5

deep-models

Implementation of a couple deep learning models using TensorFlow
Python
145
star
6

mamba.rs

Rust
121
star
7

xla-rs

Experimentation using the xla compiler from rust
Rust
87
star
8

npy-ocaml

Numpy file format support for ocaml.
OCaml
41
star
9

ocaml-arrow

OCaml
34
star
10

ocaml-rust

Safe OCaml-Rust Foreign Function Interface
Rust
34
star
11

ocaml-wasmtime

OCaml WebAssembly runtime powered by Wasmtime
OCaml
34
star
12

tch-ext

Sample Python extension using Rust/PyO3/tch to interact with PyTorch
Rust
31
star
13

ocaml-matplotlib

Plotting for ocaml based on matplotlib.pyplot
OCaml
30
star
14

btc-ocaml

A toy implementation of the bitcoin protocol in ocaml.
OCaml
29
star
15

ocaml-xla

XLA (Accelerated Linear Algebra) bindings for OCaml
OCaml
28
star
16

ocaml-dataframe

Simple and type-safe dataframe api implemented in pure ocaml
OCaml
25
star
17

ocaml-bert

Transformer-based models for Natural Language Processing in OCaml
OCaml
23
star
18

binprot-rs

Bin_prot binary protocols in Rust
Rust
19
star
19

ocaml-onnx

OCaml ONNX runtime powered by onnxruntime
C
18
star
20

ocaml-tqdm

An ocaml progress bar library similar to https://tqdm.github.io
OCaml
17
star
21

rsexp

S-expression parsing and writing in Rust
Rust
17
star
22

sphn

python bindings for symphonia/opus - read various audio formats from python and write opus files
Rust
16
star
23

tboard-rs

Read and write tensorboard data using Rust
Rust
16
star
24

ProjectEuler

Python
15
star
25

glim

Rust
15
star
26

ocaml-minipy

Naive interpreter for a Python like language
OCaml
13
star
27

syncarp

An async rpc implementation based on tokio and compatible with OCaml Async_rpc
Rust
12
star
28

ocaml.jl

Prototype code for some Julia-OCaml bindings
OCaml
12
star
29

ocaml-tensorflow-eager

OCaml bindings for TensorFlow Eager mode
OCaml
11
star
30

wtensor

Experiments around a webgpu based tensor library
Rust
9
star
31

hojo

A small python library to run iterators in a separate process
Rust
9
star
32

LaurentMazare.github.io

JavaScript
8
star
33

timens-rs

Simple and efficient time representation in Rust.
Rust
7
star
34

ocaml-smbus

C
6
star
35

cmt-fun

OCaml
6
star
36

serde-binprot

Rust binprot serialization using serde
Rust
6
star
37

ocaml-jupyter-async

An OCaml kernel for Jupyter using async.
OCaml
5
star
38

jax-flash-attn3

JAX bindings for the flash-attention3 kernels
C++
5
star
39

openai-gym-ocaml

OCaml
4
star
40

ocaml-tensorboard

Write tensorboard compatible log files from ocaml
OCaml
4
star
41

ocaml-rust-stubs

OCaml
4
star
42

ocaml-rpi-gpio

ocaml api for raspberry pi gpio access
C
3
star
43

ocaml-gym

Bindings for OpenAI Gym using the Python C API
OCaml
3
star
44

ogg-table

Ogg-vorbis reader with fast random access
Rust
3
star
45

ocaml-rplidar

RPLidar A1M8 ocaml library
OCaml
2
star
46

ocamldate

Very simple ocaml date implementation
OCaml
1
star