rtqichen

torchdiffeq

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

ffjord

code for "FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Models".

style-swap

code for "Fast Patch-based Style Transfer of Arbitrary Style".

beta-tcvae

code for "Isolating Sources of Disentanglement in Variational Autoencoders".

residual-flows

code for "Residual Flows for Invertible Generative Modeling".

torch-convgru

Dumping code for a convolutional GRU network implementation using nngraph. Allows "parameter sharing relaxation".

cudaGP

Implementation of Gaussian Process regression in CUDA C.

rtqichen.github.io