pymc-devs / pytensor

Related projects ⓘ

Alternatives and complementary repositories for pytensor

• aesara-devs / aesara
  Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arra…
  ☆1,180Updated 2 months ago
• stan-dev / pystan
  PyStan, a Python interface to Stan, a platform for statistical modeling. Documentation: https://pystan.readthedocs.io
  ☆342Updated 4 months ago
• blackjax-devs / blackjax
  BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
  ☆841Updated last week
• pyro-ppl / numpyro
  Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
  ☆2,191Updated this week
• probabilistic-numerics / probnum
  Probabilistic Numerics in Python.
  ☆438Updated 6 months ago
• pymc-devs / pymc-examples
  Examples of PyMC models, including a library of Jupyter notebooks.
  ☆288Updated last month
• jax-ml / oryx
  Oryx is a library for probabilistic programming and deep learning built on top of Jax.
  ☆219Updated 2 weeks ago
• matthewwardrop / formulaic
  A high-performance implementation of Wilkinson formulas for Python.
  ☆344Updated 3 months ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX.
  ☆459Updated last week
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆355Updated 3 weeks ago
• jax-ml / bayeux
  State of the art inference for your bayesian models.
  ☆164Updated last month
• dfm / tinygp
  The tiniest of Gaussian Process libraries
  ☆296Updated last week
• mpi4jax / mpi4jax
  Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python
  ☆441Updated last week
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆330Updated last week
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,182Updated 2 weeks ago
• google / jaxopt
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆931Updated last month
• probml / dynamax
  State Space Models library in JAX
  ☆691Updated this week
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆321Updated 6 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,689Updated last week
• probml / sts-jax
  Structural Time Series in JAX
  ☆184Updated 6 months ago
• fehiepsi / rethinking-numpyro
  Statistical Rethinking (2nd ed.) with NumPyro
  ☆446Updated 8 months ago
• pymc-devs / pymc4
  (Deprecated) Experimental PyMC interface for TensorFlow Probability. Official work on this project has been discontinued.
  ☆711Updated 3 years ago
• pymc-devs / sunode
  Solve ODEs fast, with support for PyMC
  ☆107Updated 5 months ago
• aesara-devs / aeppl
  Tools for an Aesara-based PPL.
  ☆64Updated 2 weeks ago
• scikit-hep / awkward
  Manipulate JSON-like data with NumPy-like idioms.
  ☆840Updated this week
• wilson-labs / cola
  Compositional Linear Algebra
  ☆430Updated 2 weeks ago
• patrick-kidger / diffrax
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,431Updated last week
• pymc-devs / pymc-experimental
  ☆82Updated last week
• google-deepmind / chex
  ☆785Updated 2 weeks ago
• numba / numba-scipy
  numba_scipy extends Numba to make it aware of SciPy
  ☆258Updated 2 months ago