Jit Python Packages

flashinfer-python

FlashInfer: Kernel Library for LLM Serving

5M 6K 977

flashinfer-cubin

FlashInfer: Kernel Library for LLM Serving

3.5M 6K 977

deepfriedmarshmallow

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

99K 9 0

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.

19K 6 1

pennylane-catalyst

A JIT compiler for hybrid quantum programs in PennyLane

15K 211 73

britive

Python SDK for the Britive REST API

12K 7 3

jaxsim

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

9K 196 22

pybritive

Britive CLI Python Implementation

8K 24 6

devito

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

7K 694 254

transonic

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

6K 126 0

datarax

A Differentiable Data Pipeline Framework for JAX

5K 1 0

yolort

yolort is a runtime stack for object detection on specialized accelerators.

3K 730 154

wasmtime-bin

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

3K 18K 2K

fory-compiler

A blazingly fast multi-language serialization framework for idiomatic domain objects, schema IDL, and cross-language data exchange.

3K 4K 414

hilbertsfc

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

3K 7 1

nufftax

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

2K 16 1

mpi4jax

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

2K 525 32

torch-staintools

GPU-accelerated Stain Normalization and Augmentation in PyTorch

998 43 5

py-cryptonight

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

811 23 16

justjit

A JIT compiler for Python 3.13 using LLVM ORC. Implements a custom generator state machine, SIMD auto-vectorization (AVX/SSE), and native async/await support.

703 4 5