IBM / pylspackLinks
A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
☆20Updated 4 months ago
Alternatives and similar repositories for pylspack
Users that are interested in pylspack are comparing it to the libraries listed below
Sorting:
- Stochastic Automatic Differentiation library for PyTorch.☆206Updated last year
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Riemannian Optimization Using JAX☆52Updated last year
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- ☆170Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆105Updated 5 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Updated last year
- A differentiation API for PyTorch☆30Updated 5 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Efficient Householder Transformation in PyTorch☆66Updated 4 years ago
- ☆214Updated 3 years ago
- ☆50Updated 4 years ago
- CUDA kernels for generalized matrix-multiplication in PyTorch☆85Updated 3 years ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆76Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Normalizing Flows using JAX☆84Updated last year
- Monotone operator equilibrium networks☆53Updated 5 years ago
- Large-scale, multi-GPU capable, kernel solver☆191Updated last month
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Updated 2 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated 2 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated last year
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 11 months ago
- Jax SSM Library☆49Updated 2 years ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆21Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆171Updated 3 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆189Updated 9 months ago
- ☆80Updated 4 years ago
- Random feature latent variable models in Python☆23Updated 2 years ago