fabiodimarco / torch-levenberg-marquardtLinks
PyTorch implementation of Levenberg-Marquardt training algorithm
☆65Updated 3 months ago
Alternatives and similar repositories for torch-levenberg-marquardt
Users that are interested in torch-levenberg-marquardt are comparing it to the libraries listed below
Sorting:
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆253Updated 2 years ago
- Newton and Quasi-Newton optimization with PyTorch☆354Updated last month
- A tiny library for stochastic dataset caching in PyTorch.☆43Updated last year
- NumPy arrays, ready for human consumption☆70Updated 3 weeks ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆38Updated 2 months ago
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- Library to make any existing neural network architecture equivariant☆53Updated 8 months ago
- High-order spline interpolation in PyTorch☆67Updated 10 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 11 months ago
- Differentiable signal processing on the sphere for PyTorch☆478Updated this week
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆251Updated this week
- Functional N-dimensional convolution in Pytorch, recursively calling convNd until reaching conv3d.☆79Updated 2 months ago
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆107Updated 5 months ago
- Compositional Linear Algebra☆478Updated last month
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆206Updated 7 months ago
- Cubic spline interpolation on multidimensional grids in PyTorch☆32Updated 3 months ago
- A parallel ODE solver for PyTorch☆260Updated 9 months ago
- 3D Gaussian Splatting in JAX☆62Updated last year
- TUI for the Slurm Workload Manager☆186Updated 2 weeks ago
- numerical optimization in pytorch☆139Updated last year
- Geometric kernels on manifolds, meshes and graphs☆261Updated this week
- ☆29Updated 6 months ago
- Pairwise Metrics for PyTorch☆19Updated last month
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆118Updated 4 months ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆448Updated 8 months ago
- Running Jax in PyTorch Lightning☆106Updated 7 months ago
- Run PyTorch in JAX. 🤝☆256Updated 2 weeks ago
- Memory mapped numpy arrays of varying shapes☆299Updated last year
- Diffusion models in PyTorch☆107Updated 3 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆111Updated 4 months ago