☆250Feb 20, 2026Updated last week
Alternatives and similar repositories for synjax
Users that are interested in synjax are comparing it to the libraries listed below
Sorting:
- Loopy belief propagation for factor graphs on discrete variables in JAX☆164Oct 18, 2024Updated last year
- A Python package for probabilistic state space modeling with JAX☆932Jan 6, 2026Updated last month
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆516Feb 22, 2026Updated last week
- ☆23Jul 23, 2021Updated 4 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- ☆620Jan 23, 2026Updated last month
- JAX Arrays for human consumption☆110Oct 23, 2025Updated 4 months ago
- An interactive tool for analyzing, executing, and improving dynamic programming algorithms.☆19Jan 30, 2026Updated last month
- Turn SymPy expressions into trainable JAX expressions.☆363Feb 4, 2026Updated 3 weeks ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,021Feb 3, 2026Updated 3 weeks ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- Multiple dispatch over abstract array types in JAX.☆137Dec 15, 2025Updated 2 months ago
- ☆25Sep 19, 2023Updated 2 years ago
- Numerical quadrature with JAX☆77Feb 16, 2026Updated 2 weeks ago
- Exponential families for JAX☆76Feb 20, 2026Updated last week
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,912Feb 23, 2026Updated last week
- ☆14Feb 1, 2024Updated 2 years ago
- Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing☆20Jan 9, 2026Updated last month
- ☆219Feb 21, 2026Updated last week
- Tools for all things related to Combinatory Categorial Grammar☆19Jul 12, 2025Updated 7 months ago
- Stochastic trace estimation using JAX☆17Aug 20, 2025Updated 6 months ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,743Feb 16, 2026Updated 2 weeks ago
- Optax is a gradient processing and optimization library for JAX.☆2,193Updated this week
- ☆44Jun 2, 2024Updated last year
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆544Feb 16, 2026Updated 2 weeks ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,610Feb 22, 2026Updated last week
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 20, 2026Updated last week
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆314Updated this week
- Tutorials and sampling algorithm comparisons☆80Updated this week
- X (weighted / probabilistic) Context-Free Grammars☆25Jan 30, 2024Updated 2 years ago
- Stack neural networks applied to hefty natural language tasks.☆15Dec 26, 2019Updated 6 years ago
- Implementation of fast algorithms for Maximum Spanning Tree (MST) parsing that includes fast ArcMax+Reweighting+Tarjan algorithm for sing…☆16Aug 9, 2023Updated 2 years ago
- JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"☆19Jun 10, 2023Updated 2 years ago
- Flacuna was developed by fine-tuning Vicuna on Flan-mini, a comprehensive instruction collection encompassing various tasks. Vicuna is al…☆111Sep 10, 2023Updated 2 years ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- Compositional Linear Algebra☆508Aug 1, 2025Updated 7 months ago
- Code Repository for "Please Mind the Root: Decoding Arborescences for Dependency Parsing" and "On Finding the K-best Non-projective Depen…☆20Dec 12, 2022Updated 3 years ago
- ☆16Apr 8, 2025Updated 10 months ago