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
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆194Updated 7 months ago
- A parallel ODE solver for PyTorch☆251Updated 5 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆386Updated last month
- Agustinus' very opiniated publication-ready plotting library☆62Updated 2 months ago
- This is the repository for Fair Universe Uncertainty Challenge☆12Updated this week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆26Updated last month
- A framework for composing Neural Processes in Python☆83Updated 3 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 11 months ago
- Machine Learning and the Physical World module☆28Updated 4 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆247Updated 2 years ago
- JAX Arrays for human consumption☆90Updated last year
- Gaussian processes in JAX.☆487Updated last week
- Implementation of a list sweeper: instead of the cartesian product, sweep over the zipped list☆19Updated 2 weeks ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 8 months ago
- ☆152Updated 3 weeks ago
- State Space Models library in JAX☆751Updated this week
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated 2 weeks ago
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆213Updated 9 months ago
- ☆144Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated 2 weeks ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆169Updated 3 years ago
- Mixture density network implemented in PyTorch.☆142Updated last year
- Minimal Implementation of Bayesian Optimization in JAX☆90Updated 2 months ago
- ☆154Updated last year
- Materials of the Nordic Probabilistic AI School 2022.☆175Updated 2 years ago
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆573Updated last week
- Material for the "Probabilistic Machine Learning" Course at the University of Tübingen, Summer Term 2023☆147Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆201Updated 4 months ago
- Turn SymPy expressions into trainable JAX expressions.☆332Updated 2 months ago
- Algorithms for computations on random manifolds made easier☆89Updated last year