Skip to content

Latest commit

 

History

History
28 lines (19 loc) · 842 Bytes

README.md

File metadata and controls

28 lines (19 loc) · 842 Bytes
Raw

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