Dmitrii Kochkov (@dkochkov1) profile picture

Discovering synergies between ML, computing and physics.

Cambridge, MA
Joined January 2021
Photo by Dmitrii Kochkov (@dkochkov1)Photo by Dmitrii Kochkov (@dkochkov1)Photo by Dmitrii Kochkov (@dkochkov1)Photo by Dmitrii Kochkov (@dkochkov1)
Dmitrii Kochkov retweeted
NeuralGCM was a huge team effort, and I would especially like to acknowledge the core team who has worked on the project over the past three years: @dkochkov1, @janniyuval, @langmore, @pnorgaard314, @singledispatch
2
1
35
2,723
Dmitrii Kochkov retweeted
New open source release from my team at Google: Dinosaur, a differentiable dynamical core for global atmospheric modeling, written in JAX: github.com/google-research/d… Dinosaur is a core component of NeuralGCM and we hope it is useful for the weather/climate research community.
7
59
8
417
51,964
We have a paper in ICLR! The title is “Learned Coarse Models for Efficient Turbulence Simulation.” We wanted to see if we could train general-purpose ML models to predict turbulent dynamics accurately at low spatial and temporal resolutions (1/n).
6
81
12
404
1/2 Our work "Learning ground states of quantum Hamiltonians with graph networks" is out! arxiv.org/abs/2110.06390 TLDR: we do Variational Monte Carlo on Heisenberg Hamiltonians using graph-nets as variational ansatz. w/ @spectralhippo Alvaro Sanchez @PeterWBattaglia @bkclark4
2
1
14
2/2 Key learnings: 1) Graph-nets learn amplitudes and phases that generalize to larger lattice sizes 2) Learning phases and amplitudes with separate networks gives better results 3) GNNs have easier time restoring symmetries than learning in a fully symmetric subspace
2
Dmitrii Kochkov retweeted
Can machine learning improve physics constrained optimization tasks, like those at the heart of numerical weather forecasting? Happy to share some our work on "Variational Data Assimilation with a Learned Inverse Observation Operator", tomorrow/today at ICML. 🧵
6
47
3
282
Dmitrii Kochkov retweeted
Come here our selected contributed talk on simulating turbulence with deep learning at #SimDL (simdl.github.io/) today Led by @neuro_kim, Alvaro Sanchez, @DrumBushField, Dmitrii Kochkov, @MilesCranmer, @spectralhippo, Jonathan Goodwin, Elaine Cui and @PeterWBattaglia
1
14
78
The paper “Learning Mesh-Based Simulation with Graph Networks” shows how to effectively learn physics simulation on irregular meshes. The model can simulate vastly different systems with significant speedups over the ground truth solver: dpmd.ai/2PubkYS #ICLR2021 (3/3)
16
216
30
955
1/2 Excited to share "Machine learning accelerated computational fluid dynamics" arxiv.org/pdf/2102.01010.pdf We use ML inside a CFD simulator to advance the accuracy/speed Pareto frontier with/ Jamie A. Smith Ayya Alieva Qing Wang Michael P. Brenner @shoyer
12
114
17
497
2/2 We show that our models enable accurate simulations on coarse grids achieving 40-80x speedup on the same hardware while generalizing well to flows not seen during training!
4
4
1
60