Jacob Goodale's repositories
ptx-kernels
Some side by side comparisons of simple CUDA kernels written in pure inline PTX.
annotated-s5
A tutorial on Simplified Structured State Spaces for Sequence Modeling
haiku_SSMs
Structured State-Space Models implemented in Haiku
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.
hloenv
an environment based on XLA for deep learning compiler optimization research.
ivy
The Unified Machine Learning Framework
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
memory
Differentiable memory modules for machine learning with Ivy
iree-rs
Rustic bindings for IREE
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
llama
Inference code for LLaMA models
melior
The rustic MLIR bindings in Rust
models
Off-the-shelf models for a variety of domains
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