wiseodd / pub-ready-plots
Agustinus' very opiniated publication-ready plotting library
☆61Updated 4 months ago
Alternatives and similar repositories for pub-ready-plots:
Users that are interested in pub-ready-plots are comparing it to the libraries listed below
- Matrix-free linear algebra in JAX.☆113Updated last week
- JAX Arrays for human consumption☆90Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆93Updated 8 months ago
- Riemannian Optimization Using JAX☆47Updated last year
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated last year
- Normalizing Flows using JAX☆82Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆59Updated this week
- Multiple dispatch over abstract array types in JAX.☆113Updated 2 weeks ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆121Updated 4 months ago
- Sketched matrix decompositions for PyTorch☆69Updated 8 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆40Updated 4 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆110Updated this week
- Parameter-Free Optimizers for Pytorch☆109Updated 8 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆24Updated this week
- Materials of the Nordic Probabilistic AI School 2023.☆85Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆99Updated this week
- ☆46Updated this week
- ☆49Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆101Updated 2 months ago
- ☆18Updated last year
- Code for the book "The Elements of Differentiable Programming".☆70Updated 4 months ago
- ☆129Updated 3 weeks ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆9Updated 2 years ago
- All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)☆21Updated 2 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆128Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆99Updated last year
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- ☆15Updated 4 months ago
- Pytorch-like dataloaders in JAX.☆67Updated 2 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆24Updated last month