jamielohoff / graphaxLinks
Cross-Country Elimination in JAX
☆17Updated 4 months ago
Alternatives and similar repositories for graphax
Users that are interested in graphax are comparing it to the libraries listed below
Sorting:
- Matrix-free linear algebra in JAX.☆133Updated this week
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆90Updated this week
- Website for the book "The Elements of Differentiable Programming".☆14Updated last month
- Multiple dispatch over abstract array types in JAX.☆127Updated last month
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆49Updated 4 months ago
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆10Updated 2 weeks ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆16Updated last year
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆29Updated 3 months ago
- A minimal JAX library for connectivity analysis at scales☆10Updated 3 months ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated last year
- JAX Arrays for human consumption☆106Updated last month
- Automatic Functional Differentiation in JAX☆74Updated 4 months ago
- Orthogonal polynomials with JAX☆22Updated last week
- Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence.☆57Updated this week
- A package for computing matrix exponentials and finite horizon Gramians☆11Updated last month
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆48Updated 3 weeks ago
- Stencil computations in JAX☆71Updated last year
- Physics-Enhanced Regression for Initial Value Problems☆20Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆112Updated 4 months ago
- Combinatorial optimization layers for machine learning pipelines☆126Updated 3 weeks ago
- Checkpointing for Automatic Differentiation☆56Updated this week
- Rescure useful code in TensorNetwork by google for further personal usage.☆12Updated 8 months ago
- A pure Julia implementation of denoising diffusion probabilistic models☆69Updated last year
- A Julia package for differentiating through expectations with Monte-Carlo estimates☆15Updated 8 months ago
- Implementation of normalizing flows compatible with Bijectors.jl☆42Updated this week
- Numerical quadrature with JAX☆66Updated last week
- Comparsion of Julia's GPU Kernel based ODE solvers with other open-source GPU ODE solvers☆26Updated last year
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆283Updated 2 weeks ago
- Taylor-mode automatic differentiation for higher-order derivatives☆79Updated 2 weeks ago
- Julia implementation of the Kolmogorov-Arnold network with custom gradients for fast training.☆77Updated 6 months ago