Scholar
Jiaxin Shi
Google Scholar ID: juZXbFoAAAAJ
Research Scientist, Google DeepMind
Citations & Impact
Citations
2,277
H-index
20
i10-index
23
Publications
20
Co-authors
25
list available
Publications
Browse publications on Google Scholar (top-right) ↗
Resume (English only)
Academic Achievements
- - Published multiple papers in NeurIPS 2024, ICML 2020, UAI 2019, ICML 2018, ICML 2025, ICML 2023, NeurIPS 2023, NeurIPS 2022, AISTATS 2022, ICLR 2022, IEEE TPAMI, AABI Symposium 2020, AISTATS 2020, etc.
- - Won the NeurIPS 2022 Outstanding Paper Award
- - Top reviewer for NeurIPS 2022
Research Experience
- - Research Scientist at Google DeepMind
- - Co-organizer of the ICLR 2025 Workshop on World Models
- - Gave talks at TTIC, Imperial College London, GenU 2024, FIMI 2024, Optimal Transport Berlin 2024, etc.
- - Area chair for AISTATS 2025, NeurIPS 2025, NeurIPS 2024, AISTATS 2023, NeurIPS 2023, etc.
Background
- Research Interests: Generative modeling, algorithmic modeling; Specialization: Machine learning, with a focus on variational inference, sampling, gradient estimation, and score-based modeling.
Miscellany
- Open-source project: masked diffusion models (with data and model parallel support)
