IBM / pylspack
A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
☆19Updated 5 months ago
Alternatives and similar repositories for pylspack:
Users that are interested in pylspack are comparing it to the libraries listed below
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆19Updated 7 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- Monotone operator equilibrium networks☆51Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 6 months ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆16Updated 3 years ago
- Efficient Householder Transformation in PyTorch☆64Updated 3 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆100Updated last year
- Kernel Stein Discrepancy Descent : a method to sample from unnormalized densities☆21Updated 10 months ago
- Graph Learning with JAX☆14Updated 2 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆104Updated 4 years ago
- Efficient Riemannian Optimization on Stiefel Manifold via Cayley Transform☆37Updated 5 years ago
- Neural Fixed-Point Acceleration for Convex Optimization☆29Updated 2 years ago
- Pytorch implementation of SuperPolyak subgradient method.☆43Updated 2 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆25Updated 3 years ago
- ☆49Updated 4 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Updated 4 years ago
- ☆48Updated 6 months ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Probabilistic Solution of Differential Equations☆13Updated 2 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆76Updated 8 months ago
- Python implementation of projection losses.☆25Updated 5 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆28Updated 4 months ago