ryanccarelli

ssljax

self-supervised learning in jax

notes

assorted notes

CADA-VAE-PyTorch

Pytorch implementation of the paper "Generalized Zero- and Few-Shot Learning via Aligned Variational Autoencoders" (CVPR 2019)

cuml-serialization

cuML - RAPIDS Machine Learning Library

deepvariant

DeepVariant is an analysis pipeline that uses a deep neural network to call genetic variants from next-generation DNA sequencing data.

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

dtm

This implements topics that change over time (Dynamic Topic Models) and a model of how individual documents predict that change.

esvit

EsViT: Efficient self-supervised Vision Transformers

flashinfer

FlashInfer: Kernel Library for LLM Serving

haiku-mup

A port of muP to JAX/Haiku

HamiltonianGenerativeNetworks

Implementation of Hamiltonian Generative Networks and Hamiltonian Flows, described in arxiv.org/abs/1909.13789

hdbscan

A high performance implementation of HDBSCAN clustering.

JaBbA

MIP based joint inference of copy number and rearrangement state in cancer whole genome sequence data.

nccl

Optimized primitives for collective multi-GPU communication

pytorch-vqvae

Vector Quantized VAEs - PyTorch Implementation

scenic

Scenic: A Jax Library for Computer Vision Research and Beyond

seurat

R toolkit for single cell genomics

svca

t5x

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

TrajectoryNet

TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamics

tvo

Code for the Thermodynamic Variational Objective

vq-vae-2-pytorch

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch

waveglow-vqvae

WaveGlow vocoder with VQVAE