fabiodimarco / torch-levenberg-marquardt
PyTorch implementation of Levenberg-Marquardt training algorithm
☆48Updated 3 weeks 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
- PyHopper is a hyperparameter optimizer, made specifically for high-dimensional problems arising in machine learning research.☆86Updated last year
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆44Updated 8 months ago
- Running Jax in PyTorch Lightning☆89Updated 2 months ago
- Graph neural networks in JAX.☆67Updated 8 months ago
- NumPy arrays, ready for human consumption☆66Updated 7 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- Geometric kernels on manifolds, meshes and graphs☆241Updated this week
- Run PyTorch in JAX. 🤝☆225Updated 3 weeks ago
- ☆161Updated 3 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆105Updated last month
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 6 months ago
- numerical optimization in pytorch☆137Updated 9 months ago
- Lightning-like training API for JAX with Flax☆38Updated 3 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 7 months ago
- Autoregressive Models in PyTorch.☆79Updated 2 years ago
- ☆149Updated 6 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆178Updated 5 months ago
- Use Jax functions in Pytorch☆237Updated last year
- Automatic gradient descent☆207Updated last year
- Neural Networks for JAX☆83Updated 5 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆70Updated 7 months ago
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆41Updated 4 months ago
- Uncertainty quantification with PyTorch☆344Updated this week
- ☆59Updated 3 years ago
- Newton and Quasi-Newton optimization with PyTorch☆341Updated 11 months ago
- 3D Gaussian Splatting in JAX☆57Updated 9 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆212Updated last week
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆46Updated 2 weeks ago
- Differentiable Algorithms and Algorithmic Supervision.☆113Updated last year
- ☆38Updated 2 years ago