Moksh Jain
jainmoksh.bsky.social
Moksh Jain
@jainmoksh.bsky.social
PhD student at Université de Montréal and Mila
https://mj10.github.io/
Check out our new work on action abstractions for amortized samplers led by @boussifo.bsky.social! Simple tokenization schemes like BPE result in meaningful action abstractions with several empirical implications for amortized samplers. Come chat with us @iclr-conf.bsky.social!
Ecstatic to show off some work my brilliant colleagues and I did at @iclr-conf.bsky.social this year! 🚀

We address the credit assignment challenge under long trajectories in RL or GFlowNets by constructing high order actions, or “chunks”, effectively compressing trajectory lengths!
April 5, 2025 at 1:28 AM