lezcano/geotorch
A PyTorch library providing constrained and manifold optimization utilities for deep learning parameterizations.
Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
GeoTorch enables constrained optimization on neural network parameters by providing parametrizations for common constraints like orthogonality, low-rank matrices, symmetric and skew-symmetric matrices, and sphere constraints. It integrates with standard PyTorch optimizers without requiring changes to training code. The library targets researchers and practitioners needing structured parameter spaces in deep learning models.