Stars
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
JAX - A curated list of resources https://github.com/google/jax
A library for easy and efficient manipulation of tensor networks.
Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
TensorFlow API for the Scala Programming Language
[INACTIVE] C and C++ library for STM32 ARM Cortex-M3 development boards.