Kaihua Ji's starred repositories
YouTube-simulations
Code for simulations on YouTube
torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
mathjax3-react
React component to load MathJax v3 and process raw ASCIImath and TeX formulas or HTML content.
conv_arithmetic
A technical report on convolution arithmetic in the context of deep learning
graph_nets
Build Graph Nets in Tensorflow
mlreview_notebooks
Notebooks for "A high bias low-variance introduction to Machine Learning for physicists."
code-samples
Source code examples from the Parallel Forall Blog