Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels

This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.

Static array shape checking for JAX powered by eval_shape

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

A std::execution style runtime context and High Performance RPC Transport for using OpenUCX. Including CUDA/ROCM/... devices with RDMA.

Anki is a smart spaced repetition flashcard program

Probabilistic Programming and Nested sampling in JAX

Training materials associated with NVIDIA's CUDA Training Series (www.olcf.ornl.gov/cuda-training-series/)

Code accompanying the paper "Generalized Interpolating Discrete Diffusion"

[DEPRECATED] Moved to ROCm/rocm-libraries repo

[DEPRECATED] Moved to ROCm/rocm-libraries repo

A tool and a library for bi-directional translation between SPIR-V and LLVM IR

Shared Middle-Layer for Triton Compilation

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

Minimal yet performant LLM examples in pure JAX

An implementation of PSGD Kron in JAX for distributed training in JAX or Flax

Minimal but scalable implementation of large language models in JAX

[NeurIPS 2024] Unique3D: High-Quality and Efficient 3D Mesh Generation from a Single Image

Train very large language models in Jax.

A simple library for scaling up JAX programs

jax-triton contains integrations between JAX and OpenAI Triton

Tensors and Dynamic neural networks in Python with strong GPU acceleration

The Gosub browser engine

Train transformer language models with reinforcement learning.

FastAPI framework, high performance, easy to learn, fast to code, ready for production

Everything you want to know about Google Cloud TPU