PyTorchPoly

Orthonormal polynomials using pytorch

Requirements

1. pytorch
2. pyindex

Available Polynomials

1. legendre
2. chebyshev
3. probabilist hermite

File Guide

1. example.py: shows how to build vandermonde matrices on the CPU
2. gpu_example.py: shows how to build vandermonde matrices on the GPU
3. training_example.py: shows how to train a polynomial expansion with stochastic gradient descent

Notes

Multivariate polynomials are temporarily unavailable

Author: Alex Gorodetsky
Email: goroda@umich.edu
Copyright (c) University of Michigan
License: MIT