BallisticLA / parla

Related projects ⓘ

Alternatives and complementary repositories for parla

• BallisticLA / marla
  Matlab Algorithms for Randomized Linear Algebra
  ☆14Updated last year
• BallisticLA / RandBLAS
  A header-only C++ library for sketching in randomized linear algebra
  ☆79Updated last month
• BallisticLA / RandLAPACK
  A high-performance C++ library for randomized numerical linear algebra
  ☆60Updated this week
• rfeinman / Torch-ARPACK
  Fast extremal eigensolvers for PyTorch.
  ☆17Updated 3 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆81Updated 2 weeks ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆49Updated 9 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆66Updated last year
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆24Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆50Updated 3 years ago
• PGelss / scikit_tt
  Tensor Train Toolbox
  ☆105Updated 2 weeks ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆107Updated 2 months ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆95Updated 2 months ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆121Updated 2 months ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆21Updated 6 months ago
• eepperly / XTrace
  Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"
  ☆12Updated last year
• numpi / hm-toolbox
  Toolbox for HODLR and HSS matrices in MATLAB
  ☆43Updated last year
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆57Updated 2 years ago
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆52Updated last year
• ketch / nodepy
  Analysis of initial value ODE solvers
  ☆78Updated 2 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆91Updated last year
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆65Updated 3 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆34Updated 3 weeks ago
• north-numerical-computing / anymatrix
  A MATLAB matrix collection with an interface to organize matrices
  ☆24Updated last year
• flaport / klujax
  Solve sparse linear systems in JAX using the KLU algorithm
  ☆33Updated last week
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆190Updated 4 months ago
• RehMoritz / vmc_pde
  Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…
  ☆9Updated 2 years ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆61Updated 4 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆106Updated last week
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆10Updated 3 weeks ago