Linear solvers and eigendecomposition for AWS Trainium via NKI
Neural Network Eigenvalue Estimator for quantum oscillator problem.