pymc-devs / pytensor
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
☆409Updated this week
Alternatives and similar repositories for pytensor:
Users that are interested in pytensor are comparing it to the libraries listed below
- Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…☆1,193Updated 2 months ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆875Updated last week
- PyStan, a Python interface to Stan, a platform for statistical modeling. Documentation: https://pystan.readthedocs.io☆346Updated 6 months ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,375Updated this week
- Examples of PyMC models, including a library of Jupyter notebooks.☆306Updated last week
- State Space Models library in JAX☆730Updated this week
- Probabilistic Numerics in Python.☆448Updated 8 months ago
- Gaussian processes in JAX.☆474Updated 2 months ago
- A high-performance implementation of Wilkinson formulas for Python.☆362Updated 3 weeks ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆407Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆943Updated 4 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆370Updated this week
- Optax is a gradient processing and optimization library for JAX.☆1,776Updated this week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆241Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,308Updated this week
- The tiniest of Gaussian Process libraries☆301Updated last week
- State of the art inference for your bayesian models.☆196Updated last month
- Powerful add-ons for PyMC☆93Updated this week
- Structural Time Series in JAX☆187Updated 8 months ago
- ☆812Updated 2 weeks ago
- Statistical Rethinking (2nd ed.) with NumPyro☆450Updated 11 months ago
- Tools for an Aesara-based PPL.☆64Updated 3 months ago
- Exploring and eliciting probability distributions☆108Updated this week
- A library that allows for inference on probabilistic models☆265Updated last year
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,212Updated this week
- (Deprecated) Experimental PyMC interface for TensorFlow Probability. Official work on this project has been discontinued.☆713Updated 3 years ago
- Turn SymPy expressions into trainable JAX expressions.☆327Updated this week
- Python wrapper for nuts-rs☆132Updated this week
- Exploratory analysis of Bayesian models with Python☆1,641Updated 3 weeks ago
- ☆545Updated last month