Assistant Professor @ Tsinghua CollegeAI. Previously, Postdoc @ Stanford University, PhD @ MIT, BS @ Peking University
pykan. Kolmogorov Arnold Networks
16kgrow-crystals. Getting crystal-like representations with harmonic loss
195BIMT. Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
177Omnigrok. Omnigrok: Grokking Beyond Algorithmic Data
65newton-kepler. Understand what physics/algorithms do transformers learn internally when trained on planetary motion
43physics_of_skill_learning. We study toy models of skill learning.
34schrodinger-pca. Schrodinger Principal Component Analysis
23aipoincare. Counting the number of conservation laws from trajectory data
22NNPhD-github. Code for "Machine-Learning Non-Conservative Dynamics for New-Physics Detection" (arXiv: 2106.00026)
14KindXiaoming.github.io. website
11QHMC. Quantum-inspired Hamiltonian Monte Carlo code
11sid. discovering interpretable conservation laws from differential equations
9aipoincare_2.0. AI Poincare 2.0
8entropy-estimator. Jupyter Notebook
4ml_hiddensym. Machine Learning hidden symmetries
4modded-nanogpt. NanoGPT (124M) in 3 minutes
3adam. analyze optimizer properties
3stanford-tensorflow-tutorials. This repository contains code examples for the Stanford's course: TensorFlow for Deep Learning Research.
3scaling-law-general. Jupyter Notebook
1models. Models and examples built with TensorFlow
1spatial_transformer_networks. Implementation of spatial transformer networks in keras 2.0 using tensorflow 1.0 as backend.
1Expectation_Maximization. expectation maximization
1Scaling-law-kerr. Jupyter Notebook
1relu_network. Jupyter Notebook
1