Jit Python Packages

flashinfer-python

FlashInfer: Kernel Library for LLM Serving

flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

deepfriedmarshmallow

A plug-and-play JIT implementation for Marshmallow to speed up data serialization and deserialization

jaxsim

A differentiable physics engine and multibody dynamics library for control and robot learning.

rocm-llvm-python

Low-level Cython and Python bindings to the (ROCm) LLVM and AMD COMGR C API. Also ships the official LLVM Clang bindings.

pennylane-catalyst

A JIT compiler for hybrid quantum programs in PennyLane

pybritive

Britive CLI Python Implementation

britive

Python SDK for the Britive REST API

devito

DSL and compiler framework for automated finite-differences and stencil computation

transonic

:rocket: Make your Python code fly at transonic speeds!

wasmtime-bin

A lightweight WebAssembly runtime that is fast, secure, and standards-compliant

datarax

A Differentiable Data Pipeline Framework for JAX

fory-compiler

A blazingly fast multi-language serialization framework powered by JIT and zero-copy.

justjit

Fast Python JIT compiler using LLVM ORC - lowers Python bytecode to native machine code

mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

hilbertsfc

Ultra-fast 2D & 3D Hilbert curve kernels in Python. JIT compiled, branchless, L1-cache-friendly lookup tables, loop unrolling, SIMD, and multi-threading.

py-cryptonight

Python Cryptonight binding / extension. Monero hash function, proof-of-work, cn_slow_hash()

torch-staintools

GPU-accelerated Stain Normalization and Augmentation in PyTorch

aheui

Aheui compiler & assembler toolkit.

charonload

Develop C++/CUDA extensions with PyTorch like Python scripts

nufftax

Pure JAX implementation of the Non-Uniform Fast Fourier Transform (NUFFT)

tmmax

A fast transfer matrix method written in jax for modelling optical multilayer thin films

tiny-vectordb

A tiny jit vector database with minimal dependencies and maximum performance.