pymc-devs / pytensorLinks
PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
☆493Updated this week
Alternatives and similar repositories for pytensor
Users that are interested in pytensor are comparing it to the libraries listed below
Sorting:
- Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…☆1,211Updated 7 months ago
- PyStan, a Python interface to Stan, a platform for statistical modeling. Documentation: https://pystan.readthedocs.io☆355Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆931Updated 2 months ago
- Probabilistic Numerics in Python.☆452Updated last week
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,482Updated last month
- A high-performance implementation of Wilkinson formulas for Python.☆394Updated 6 months ago
- Examples of PyMC models, including a library of Jupyter notebooks.☆339Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆457Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆269Updated this week
- A Python package for probabilistic state space modeling with JAX☆837Updated 2 months ago
- Gaussian processes in JAX and Flax.☆521Updated 2 weeks ago
- The tiniest of Gaussian Process libraries☆311Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆437Updated 3 weeks ago
- State of the art inference for your bayesian models.☆219Updated 2 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,932Updated last week
- Exploring and eliciting probability distributions☆139Updated 3 weeks ago
- numba_scipy extends Numba to make it aware of SciPy☆274Updated 10 months ago
- Turn SymPy expressions into trainable JAX expressions.☆343Updated 2 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆979Updated 2 months ago
- Exploratory analysis of Bayesian models with Python☆1,704Updated this week
- RFC document, tooling and other content related to the array API standard☆242Updated 2 weeks ago
- Composable Function Transformations in Python with Mojo/MAX acceleration☆257Updated last week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,463Updated 2 months ago
- Tools for an Aesara-based PPL.☆64Updated 8 months ago
- Manipulate JSON-like data with NumPy-like idioms.☆894Updated this week
- A library that allows for inference on probabilistic models☆271Updated last year
- ☆864Updated last week
- Powerful add-ons for PyMC☆106Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,418Updated this week
- Structural Time Series in JAX☆193Updated last year