jpgoodale Goto Github PK

ai_on_threads

annotated-s5

A tutorial on Simplified Structured State Spaces for Sequence Modeling

audiocraft

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable music generation LM with textual and melodic conditioning.

builder

Build custom Ivy training pipelines for any deep learning framework with clear, hierarchical and robust user specifications

candle

Minimalist ML framework for Rust

cudarc

Safe rust wrapper around CUDA toolkit

cutlass

CUDA Templates for Linear Algebra Subroutines

database

A collection of different algorithms, tools, repos, products, related to different layers of the AI deployment stack

demucs

Code for the paper Hybrid Spectrogram and Waveform Source Separation

encodec

State-of-the-art deep learning based audio codec supporting both mono 24 kHz audio and stereo 48 kHz audio.

haiku_cfc

Simple Haiku implementation of Closed-form Continuous-time Neural Networks

haiku_ssms

Structured State-Space Models implemented in Haiku

hippox

High-order Polynomial Projection Operators for JAX

hloenv

an environment based on XLA for deep learning compiler optimization research.

iree-jax

iree-rs

Rustic bindings for IREE

ivy

The Unified Machine Learning Framework

jax

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

jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

jpgoodale

Config files for my GitHub profile.

llama

Inference code for LLaMA models

melior

The rustic MLIR bindings in Rust

memory

Differentiable memory modules for machine learning with Ivy

models

Off-the-shelf models for a variety of domains

ptx-kernels

Some side by side comparisons of simple CUDA kernels written in pure inline PTX.

s5

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

xla

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

xla-rs

Experimentation using the xla compiler from rust

xla_binaries

Pre-compiled XLA extension