Alternatives and similar repositories for cr-sparse

Users that are interested in cr-sparse are comparing it to the libraries listed below

• v0lta / Jax-Wavelet-Toolbox
  Differentiable and gpu enabled fast wavelet transforms in JAX.
  ☆45Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Updated last year
• PyLops / pyproximal
  PyProximal – Proximal Operators and Algorithms in Python
  ☆76Updated 3 weeks ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated 2 years ago
• pmelchior / proxmin
  Proximal optimization in pure python
  ☆118Updated 3 years ago
• cvxgrp / signal-decomposition
  A simple and general framework for signal decomposition
  ☆71Updated 9 months ago
• pnkraemer / million-dimension-prob-ode-solver-experiments
  ☆10Updated 3 years ago
• ucl-bug / jwave
  A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
  ☆183Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 9 months ago
• lindermanlab / ssm-jax
  Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
  ☆60Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆98Updated 6 months ago
• mathurinm / andersoncd
  This code is no longer maintained. The codebase has been moved to https://github.com/scikit-learn-contrib/skglm. This repository only ser…
  ☆18Updated 2 years ago
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆209Updated 2 months ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated 2 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆75Updated last week
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆103Updated last year
• joaquimcampos / DeepSplines
  Learning Activation Functions in Deep (Spline) Neural Networks
  ☆28Updated 2 years ago
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆33Updated 3 years ago
• AndreiB137 / EinFields
  Einstein Fields official implementation in JAX.
  ☆61Updated 2 months ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆102Updated 2 years ago
• SamDuffield / mocat
  All things Monte Carlo, written in JAX.
  ☆32Updated 2 years ago
• EEA-sensors / sequential-parallelization-examples
  This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
  ☆33Updated 2 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆103Updated last week
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated last week
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 3 months ago
• Sohl-Dickstein / Hamiltonian-Annealed-Importance-Sampling
  Matlab code implementing Hamiltonian Annealed Importance Sampling for importance weight, partition function, and log likelihood estimatio…
  ☆26Updated 11 years ago
• IPL-UV / jaxkern
  A lightweight didactic library of kernel methods using the back-end JAX.
  ☆12Updated 2 years ago
• rfeinman / pytorch-lasso
  L1-regularized least squares with PyTorch
  ☆69Updated 2 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago