Xi Wang's repositories
torch_SB_VAE
Implementation of stick breaking VAE with pytorch
ResNet-VAE
VAE with ResNet encoder and decoder
ShiftMatchImageNet
ShfitMatch experiments on ImageNet
incubator-mxnet
Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
plotting_snippet
Some common code snippets I used for making plot
Bayesian-Neural-Networks
Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
blackjax
BlackJAX is a sampling library designed for ease of use, speed and modularity.
bnn_priors
Code for the paper "Bayesian Neural Network Priors Revisited"
coq_last_hw
Last homework for functional programming
final-assignment-open-source
华师大开源设计与开发课程期末作业
generax
Generative models with JAX.
hlb-gpt
Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wikitext-103 on a single A100 in <100 seconds. Scales to larger models with one parameter change (feature currently in alpha).
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
outlier-exposure
Deep Anomaly Detection with Outlier Exposure (ICLR 2019)
pangolin
probabilistic programming focused on fun
posteriordb
Database with posteriors of interest for Bayesian inference
pyro
Deep universal probabilistic programming with Python and PyTorch
viabel
Efficient, lightweight, variational inference approximation bounds