Zekun Shi's starred repositories
Megatron-LM
Ongoing research training transformer models at scale
DeepH-pack
Deep neural networks for density functional theory Hamiltonian.
abacus-develop
An electronic structure package based on either plane wave basis or numerical atomic orbitals.
latex-beamer-pure-minimalistic
A true minimalistic LaTeX beamer template
jax-finufft
JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
beamerthemesimple
A minimalist Beamer theme with a watermark in the background
goertzel-fft
A benchmark for Goertzel algorithm and `scipy.fftpack.fft` (with custom implementation of Goertzel algorithm in short-time version) **for evaluating a few DFT terms**
papers-101
Implementation of papers in 101 lines of code.
flows-on-sphere
This is a Pytorch implementation of [normalizing flows on tori and spheres, ICML 2020]