getkeops / keops
KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows
☆1,054Updated this week
Related projects ⓘ
Alternatives and complementary repositories for keops
- Constrained optimization toolkit for PyTorch☆661Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,396Updated this week
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆561Updated this week
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,302Updated 2 months ago
- Geometric loss functions between point clouds, images and volumes☆595Updated 10 months ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆863Updated last month
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,408Updated 3 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆933Updated 2 months ago
- Fast, differentiable sorting and ranking in PyTorch☆774Updated 10 months ago
- Profiling and inspecting memory in pytorch☆1,020Updated 3 months ago
- Fast Differentiable Sorting and Ranking☆574Updated 9 months ago
- Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.☆526Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,219Updated this week
- A PyTorch implementation of L-BFGS.☆586Updated last year
- A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods☆1,399Updated 6 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,699Updated last week
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,011Updated last week
- Riemannian Adaptive Optimization Methods with pytorch optim☆851Updated 6 months ago
- JAX - A curated list of resources https://github.com/google/jax☆1,562Updated 4 months ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆473Updated 2 years ago
- ☆788Updated this week
- Laplace approximations for Deep Learning.☆471Updated this week
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆301Updated 3 weeks ago
- PIX is an image processing library in JAX, for JAX.☆389Updated last week
- Newton and Quasi-Newton optimization with PyTorch☆320Updated 8 months ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- TensorDict is a pytorch dedicated tensor container.☆840Updated this week
- A Graph Neural Network Library in Jax☆1,376Updated 8 months ago
- A 100x faster SVD for PyTorch⚡️☆447Updated 2 years ago