LaurentMazare

tch-rs

Rust bindings for the C++ api of PyTorch.

diffusers-rs

An implementation of the diffusers api in Rust

ocaml-torch

OCaml bindings for PyTorch

mamba.rs

xla-rs

Experimentation using the xla compiler from rust

npy-ocaml

Numpy file format support for ocaml.

ocaml-rust

Safe OCaml-Rust Foreign Function Interface

ocaml-xla

XLA (Accelerated Linear Algebra) bindings for OCaml

tch-ext

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

binprot-rs

Bin_prot binary protocols in Rust

ocaml-onnx

OCaml ONNX runtime powered by onnxruntime

rsexp

S-expression parsing and writing in Rust

tboard-rs

Read and write tensorboard data using Rust

syncarp

An async rpc implementation based on tokio and compatible with OCaml Async_rpc

wtensor

Experiments around a webgpu based tensor library

LaurentMazare.github.io

hojo

A small python library to run iterators in a separate process

timens-rs

Simple and efficient time representation in Rust.

gemm

opam-repository

Main public package repository for OPAM, the source package manager of OCaml.

ml-engineering

Machine Learning Engineering Guides and Tools

are-we-learning-yet

How ready is Rust for Machine Learning?

ggblas

blas

Wrappers for BLAS (Fortran)

blas-sys

Bindings to BLAS (Fortran)

cudarc

Safe rust wrapper around CUDA toolkit

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

lit-llama

Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

ocaml-rs

OCaml extensions in Rust

safetensors

Simple, safe way to store and distribute tensors