An Application of the Holonomic Gradient Method to the Neural Tangent Kernel (for extended abstract)


  1. Codes of the full paper (computer algebra codes are here).
  2. Codes (tar.gz) of "Experiments 2" of the extended abstract (mnist letters of 0 and 1).
    It runs on the docker image (GPU is requred)
    nvcr.io/nvidia/pytorch      24.10-py3 
    

    tar xzvf as-dist.tar.gz
    cd as-dist
    python experiment_mnist_hgm.py
       (hgm)
    python experiment_mnist_other.py  
       (closed formula, gauss-hermite quadrature, Monte-Carlo)