An efficient C++17 GPU numerical computing library with Python-like syntax

The simplest, fastest repository for training/finetuning medium-sized GPTs.

Kernel Tuner

Productive, portable, and performant GPU programming in Python.

A high-performance, zero-overhead, extensible Python compiler with built-in NumPy support

PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.

Code for the "Unified Modulated Pattern Analysis" (UMPA) method of processing speckle-based X-ray imaging data

Lazily evaluated arrays in Python

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

Modern scheduling library for Python

Computational Tomography in PyTorch

10x faster matrix and vector operations

High-Resolution Image Synthesis with Latent Diffusion Models

A library for painless 3D tomographic reconstruction

Learning in infinite dimension with neural operators.

A deblocking JPEG decoder

Mathematical operations for JAX pytrees

A Pytree Module system for Deep Learning in JAX

An Aspiring Drop-In Replacement for NumPy at Scale

This repository contains the source code for the paper First Order Motion Model for Image Animation