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.
  ☆42Updated last year
• cvxgrp / signal-decomposition
  A simple and general framework for signal decomposition
  ☆68Updated 6 months ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆128Updated 10 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• liuyao12 / ConvNets-PDE-perspective
  an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets
  ☆22Updated last year
• pnkraemer / million-dimension-prob-ode-solver-experiments
  ☆10Updated 3 years ago
• pmelchior / proxmin
  Proximal optimization in pure python
  ☆118Updated 3 years ago
• PyLops / pyproximal
  PyProximal – Proximal Operators and Algorithms in Python
  ☆70Updated last week
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆32Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆100Updated 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
• EEA-sensors / sequential-parallelization-examples
  This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
  ☆32Updated last year
• AaltoML / PeriodicBNN
  Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)
  ☆19Updated 3 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 6 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated 2 months ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆97Updated last year
• 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
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆126Updated last month
• esa / torchquad
  Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX
  ☆205Updated 7 months ago
• lindermanlab / ssm-jax
  Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
  ☆61Updated last year
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 5 years ago
• aai-institute / continuiti
  Learning function operators with neural networks.
  ☆32Updated 10 months ago
• ucl-bug / jwave
  A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
  ☆171Updated 10 months ago
• cvxgrp / a2dr
  Anderson accelerated Douglas-Rachford splitting
  ☆29Updated 4 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆72Updated 3 weeks ago
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆95Updated 2 months ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆97Updated last week
• AdrienCorenflos / particle_mala
  Gradient-informed particle MCMC methods
  ☆12Updated last year