pearu / functional_algorithmsLinks
Functional algorithms - definitions and implementations
☆12Updated last month
Alternatives and similar repositories for functional_algorithms
Users that are interested in functional_algorithms are comparing it to the libraries listed below
Sorting:
- Use numba-compiled kernels from within Jax☆31Updated 4 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆96Updated this week
- Exponential families for JAX☆75Updated this week
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Stencil computations in JAX☆71Updated last year
- Generic API for dispatch to Pyro backends.☆16Updated 3 years ago
- Linnea is an experimental tool for the automatic generation of optimized code for linear algebra problems.☆71Updated last month
- A dead simple Python package for creating custom JAX pytree objects☆64Updated 10 months ago
- ONNX-backed array library that is compliant with the Array API standard.☆59Updated 3 weeks ago
- AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.☆40Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆40Updated 4 months ago
- Python bindings for the egglog rust library☆89Updated this week
- Compiler with automatic differentiation☆48Updated last year
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆63Updated last year
- Documentation:☆122Updated 2 years ago
- LattE integrale, software for counting lattice points and integration over convex polytopes☆48Updated 4 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆130Updated last year
- Tensor-like types – with variadic shapes – that support both static and runtime type checking, and convenient parsing☆19Updated 3 weeks ago
- einsum via batch matrix multiply☆13Updated last year
- Mathematical operations for JAX pytrees☆200Updated 9 months ago
- Bayesian optimization of discrete sequences☆21Updated 3 years ago
- An HMC/NUTS implementation in Aesara☆31Updated 2 years ago
- Python S-expression emulation using tuple-like objects.☆22Updated 2 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆97Updated 5 months ago
- Tools for JAX☆49Updated this week
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- ∂B nets: learning discrete, boolean-valued functions by gradient descent☆21Updated last year
- Multiple dispatch over abstract array types in JAX.☆129Updated last week
- Creates performance portable libraries with embedded source representations.☆26Updated 9 months ago