anthonyhu / ml-research
Codebase for machine learning research in PyTorch.
☆13Updated 2 years ago
Alternatives and similar repositories for ml-research:
Users that are interested in ml-research are comparing it to the libraries listed below
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆687Updated 2 months ago
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆26Updated last week
- A simple pytorch implementation for calculating VAE loss components and annealing KLD loss while training VAEs, especially RNN-based☆18Updated 4 months ago
- ☆565Updated last week
- This is the repository for Fair Universe Uncertainty Challenge☆12Updated 3 weeks ago
- A parallel ODE solver for PyTorch☆254Updated 6 months ago
- Gaussian processes in JAX and Flax.☆504Updated last week
- CLU lets you write beautiful training loops in JAX.☆337Updated 2 weeks ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆580Updated this week
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆578Updated 3 months ago
- JAX Arrays for human consumption☆92Updated last year
- Example of how to use Weights & Biases on Slurm☆114Updated 2 years ago
- Add a tqdm progress bar to your JAX scans and loops.☆106Updated last month
- A framework for composing Neural Processes in Python☆85Updated 4 months ago
- Probabilistic Numerics in Python.☆448Updated 11 months ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆196Updated 8 months ago
- A minimalist cookiecutter for research and data science projects☆31Updated last year
- Running Jax in PyTorch Lightning☆94Updated 4 months ago
- Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container☆33Updated 2 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆99Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆218Updated 10 months ago
- ☆157Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆94Updated last week
- Repository of the journal club "Diffusion Models and Generative Modeling"☆17Updated 5 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆367Updated 2 weeks ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 9 months ago
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆201Updated last year
- Code for our NeurIPS 2022 paper☆367Updated 2 years ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆251Updated 2 years ago