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.
  ☆48Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Updated 3 weeks ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated 2 months ago
• PyLops / pyproximal
  PyProximal – Proximal Operators and Algorithms in Python
  ☆80Updated 2 weeks ago
• pierreablin / qndiag
  Quasi-Newton algorithm for joint-diagonalization
  ☆34Updated 3 years ago
• pnkraemer / million-dimension-prob-ode-solver-experiments
  ☆10Updated 3 years ago
• pmelchior / proxmin
  Proximal optimization in pure python
  ☆118Updated 3 years ago
• liuyao12 / ConvNets-PDE-perspective
  an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets
  ☆22Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated last year
• ucl-bug / jwave
  A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs
  ☆193Updated last year
• aai-institute / continuiti
  Learning function operators with neural networks.
  ☆35Updated last year
• AaltoML / PeriodicBNN
  Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)
  ☆19Updated 4 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆76Updated last week
• 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 3 years ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆107Updated 2 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• nlesc-dirac / pytorch
  Improved LBFGS and LBFGS-B optimizers in PyTorch.
  ☆67Updated last year
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆59Updated 2 years ago
• rfeinman / pytorch-lasso
  L1-regularized least squares with PyTorch
  ☆72Updated 2 years ago
• lindermanlab / ssm-jax
  Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
  ☆63Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 3 years ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆103Updated 2 years ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆114Updated last week
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated 2 years ago
• thoglu / jammy_flows
  A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…
  ☆45Updated 10 months ago
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆102Updated 9 months ago
• AdrienCorenflos / particle_mala
  Gradient-informed particle MCMC methods
  ☆12Updated 2 years ago
• astro-informatics / s2wav
  Differentiable and accelerated wavelet transform on the sphere with JAX
  ☆25Updated 8 months 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
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆57Updated 2 years ago