Find your next starting point.

Neural-network transformations for JAX, supporting studies of parameter initialization and functional apply behavior.

Solvers for linear, mixed-integer, and quadratic optimization problems.

Array transformations and automatic differentiation for examining compilation, vectorization and functional numerical programs.

Neural-network APIs for examining model behavior across declared computation backends.

A reference for bounded nonlinear least squares, including Jacobians, robust losses, scaling and termination behavior.

Nonlinear model fitting with named parameters, constraints, and uncertainty-analysis tools.

Derivative-free optimization methods and benchmark interfaces.

A common interface to nonlinear optimization algorithms, with and without derivatives.

Language-model research training repository for studying configuration transparency and reproducibility of model-building workflows.

Gradient transformation and optimization tools for testing optimizer state and update equations.

A solver-selection guide spanning local and global optimization, roots, constrained problems, linear programming and assignment.

Automated hyperparameter search with trial management, sampling, and pruning methods.