anthonyhu / ml-research
Codebase for machine learning research in PyTorch.
☆13Updated last year
Alternatives and similar repositories for ml-research:
Users that are interested in ml-research are comparing it to the libraries listed below
- Aggregate multiple tensorboard runs to new summary or csv files☆169Updated last year
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆237Updated 2 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆192Updated last month
- From search engines, to science, to robotics, this reposity is meant to showcase the use of reinforcement learning in the world..☆221Updated 7 months ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆188Updated 5 months ago
- State Space Models library in JAX☆726Updated last month
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆366Updated 2 weeks ago
- Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.☆555Updated 2 weeks ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆569Updated 2 weeks ago
- PyTorch implementation of bayesian neural network [torchbnn]☆506Updated 5 months ago
- Gaussian processes in JAX.☆473Updated last month
- ☆73Updated 5 months ago
- Agustinus' very opiniated publication-ready plotting library☆61Updated 4 months ago
- ☆217Updated 8 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆94Updated 9 months ago
- Repository of the journal club "Diffusion Models and Generative Modeling"☆16Updated last month
- Minimal Implementation of Bayesian Optimization in JAX☆88Updated last week
- Laplace approximations for Deep Learning.☆484Updated last month
- Materials of the Nordic Probabilistic AI School 2023.☆85Updated last year
- A parallel ODE solver for PyTorch☆244Updated 3 months ago
- A framework for composing Neural Processes in Python☆79Updated last month
- This is the repository for Fair Universe Uncertainty Challenge☆12Updated last week
- ☆276Updated last week
- A Bayesian optimization toolbox built on TensorFlow☆226Updated this week
- Compositional Linear Algebra☆456Updated last week
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆110Updated this week
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆169Updated 2 years ago
- CLU lets you write beautiful training loops in JAX.☆329Updated this week
- ☆545Updated last month
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆99Updated this week