Topographic Map Viewer

jax-triton contains integrations between JAX and OpenAI Triton

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

Web pages

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

The pythoncapi-compat project can be used to write a C extension supporting a wide range of Python versions with a single code base.

A microbenchmark support library

Computation using data flow graphs for scalable machine learning

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.

No repository description provided.

Seamless operability between C++11 and Python

Flax is a neural network library for JAX that is designed for flexibility.

No repository description provided.

A machine learning compiler for GPUs, CPUs, and ML accelerators

The Python programming language

The fundamental package for scientific computing with Python.

nanobind — Seamless operability between C++17 and Python

MLIR-based partitioning system

Run Python tests in parallel with free threading

No repository description provided.

Backward compatible ML compute opset inspired by HLO/MHLO

Supernodal sparse direct solver. https://portal.nersc.gov/project/sparse/superlu/

Optimized primitives for collective multi-GPU communication

Lingvo

Bloaty: a size profiler for binaries

FlatBuffers: Memory Efficient Serialization Library

Collective communications library with various primitives for multi-machine training.

Development repository for the Triton language and compiler

No repository description provided.

A conda-smithy repository for jaxlib.

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

Fast numerical array expression evaluator for Python, NumPy, PyTables, pandas, bcolz and more

No repository description provided.

Demonstration of pyhf Issue 1501

OpenBLAS is an optimized BLAS library based on GotoBLAS2 1.13 BSD version.

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

The Legion Parallel Programming System