A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.

Minimal, lightweight JAX implementations of popular models.

Open-source framework for the research and development of foundation models.

xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerators such as TPUs and GPUs on GKE.

A playbook for systematically maximizing the performance of deep learning models.

A retargetable MLIR-based machine learning compiler and runtime toolkit.

jax-triton contains integrations between JAX and OpenAI Triton

Rax is a Learning-to-Rank library written in JAX.

Lingvo

JAX - A curated list of resources https://github.com/google/jax

A library for easy and efficient manipulation of tensor networks.

Concise deep learning for JAX

Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf

My first GitHub attempt

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

TensorFlow library for Clojure

TensorFlow API for the Scala Programming Language

Multiple cursors for emacs.

Emacs Interface to Ack-like Tools

[INACTIVE] C and C++ library for STM32 ARM Cortex-M3 development boards.

Apache ZooKeeper quorum configuration generator