Moksh Jain
Scholar

Moksh Jain

Google Scholar ID: cGHr9_MAAAAJ
Mila, Université de Montréal
probabilistic machine learning
Homepage↗Google Scholar↗
Citations & Impact
All-time
Citations
2,102
 
H-index
20
 
i10-index
21
 
Publications
20
 
Co-authors
0
 
Contact
No contact links provided.
Publications
8 items
A Comedy of Estimators: On KL Regularization in RL Training of LLMs
2025
Cited
0
Benchmarking World-Model Learning
2025
Cited
0
Recursive Self-Aggregation Unlocks Deep Thinking in Large Language Models
2025
Cited
0
Search-Based Correction of Reasoning Chains for Language Models
2025
Cited
0
Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training
2025
Cited
0
Solving Bayesian inverse problems with diffusion priors and off-policy RL
2025
Cited
0
Amortizing intractable inference in diffusion models for vision, language, and control
arXiv.org · 2024
Cited
14
Learning diverse attacks on large language models for robust red-teaming and safety tuning
arXiv.org · 2024
Cited
9
Resume (English only)
Co-authors
0 total
Co-authors: 0 (list not available)