SinaAlemohammad / RNTK_UCI

code for RNTK on UCI dataset

RNTK_UCI

This is an efficient GPU implementation of recurrent neural tangent kernel (RNTK) for UCI dataset.

dependencies

jax

symjax

sklearn

reproduce results

open a terminal cd $PROJECT where $PROJECT is the repo location and then simply run bash main.sh.

reference

If you find this work helpful to you, please consider citing the folowing:

@misc{alemohammad2020scalable,
      author = {{Alemohammad}, Sina and {Balestriero}, Randall and {Wang}, Zichao and {Baraniuk}, Richard},
        title = "{Scalable Neural Tangent Kernel of Recurrent Architectures}",
      journal = {arXiv e-prints},
         year = 2020,
        month = dec,
       eprint = {2012.04859},
}