Follow
Moksh Jain
Moksh Jain
Mila, Université de Montréal
Verified email at mila.quebec - Homepage
Title
Cited by
Cited by
Year
Flow network based generative models for non-iterative diverse candidate generation
E Bengio, M Jain, M Korablyov, D Precup, Y Bengio
Advances in Neural Information Processing Systems 34, 27381-27394, 2021
2892021
DROCC: Deep robust one-class classification
S Goyal, A Raghunathan, M Jain, HV Simhadri, P Jain
International Conference on Machine Learning, 3711-3721, 2020
1942020
Biological Sequence Design with GFlowNets
M Jain, E Bengio, A Hernandez-Garcia, J Rector-Brooks, BFP Dossou, ...
International Conference on Machine Learning, 9786-9801, 2022
1592022
Trajectory balance: Improved credit assignment in gflownets
N Malkin, M Jain, E Bengio, C Sun, Y Bengio
Advances in Neural Information Processing Systems 35, 5955-5967, 2022
1362022
Deup: Direct epistemic uncertainty prediction
S Lahlou*, M Jain*, H Nekoei, VI Butoi, P Bertin, J Rector-Brooks, ...
Transactions on Machine Learning Research, 2023
1052023
Multi-objective gflownets
M Jain, SC Raparthy, A Hernández-Garcıa, J Rector-Brooks, Y Bengio, ...
International Conference on Machine Learning, 14631-14653, 2023
662023
Learning GFlowNets from partial episodes for improved convergence and stability
K Madan, J Rector-Brooks, M Korablyov, E Bengio, M Jain, AC Nica, ...
International Conference on Machine Learning, 23467-23483, 2023
642023
GFlowNets for AI-driven scientific discovery
M Jain, T Deleu, J Hartford, CH Liu, A Hernandez-Garcia, Y Bengio
Digital Discovery 2 (3), 557-577, 2023
442023
EdgeML: Machine Learning for resource-constrained edge devices
DK Dennis, Y Gaurkar, S Gopinath, S Goyal, C Gupta, M Jain, S Jaiswal, ...
URL https://github. com/Microsoft/EdgeML, 2020
442020
Amortizing intractable inference in large language models
EJ Hu*, M Jain*, E Elmoznino, Y Kaddar, G Lajoie, Y Bengio, N Malkin
The Twelfth International Conference on Learning Representations, 2024
412024
GFlowNet-EM for learning compositional latent variable models
EJ Hu, N Malkin, M Jain, KE Everett, A Graikos, Y Bengio
International Conference on Machine Learning, 13528-13549, 2023
392023
Stochastic generative flow networks
L Pan, D Zhang, M Jain, L Huang, Y Bengio
Uncertainty in Artificial Intelligence, 1628-1638, 2023
232023
Evaluating Generalization in GFlowNets for Molecule Design
AC Nica, M Jain, E Bengio, CH Liu, M Korablyov, MM Bronstein, Y Bengio
ICLR2022 Machine Learning for Drug Discovery, 2022
222022
Thompson sampling for improved exploration in GFlowNets
J Rector-Brooks, K Madan, M Jain, M Korablyov, CH Liu, S Chandar, ...
arXiv preprint arXiv:2306.17693, 2023
192023
Gflowout: Dropout with generative flow networks
D Liu, M Jain, BFP Dossou, Q Shen, S Lahlou, A Goyal, N Malkin, ...
International Conference on Machine Learning, 21715-21729, 2023
162023
PhyloGFN: Phylogenetic inference with generative flow networks
M Zhou, Z Yan, E Layne, N Malkin, D Zhang, M Jain, M Blanchette, ...
arXiv preprint arXiv:2310.08774, 2023
152023
Pre-Training and Fine-Tuning Generative Flow Networks
L Pan, M Jain, K Madan, Y Bengio
arXiv preprint arXiv:2310.03419, 2023
102023
Multi-Fidelity Active Learning with GFlowNets
A Hernandez-Garcia, N Saxena, M Jain, CH Liu, Y Bengio
arXiv preprint arXiv:2306.11715, 2023
102023
Amortizing intractable inference in diffusion models for vision, language, and control
S Venkatraman, M Jain, L Scimeca, M Kim, M Sendera, M Hasan, L Rowe, ...
arXiv preprint arXiv:2405.20971, 2024
82024
Learning diverse attacks on large language models for robust red-teaming and safety tuning
S Lee, M Kim, L Cherif, D Dobre, J Lee, SJ Hwang, K Kawaguchi, G Gidel, ...
arXiv preprint arXiv:2405.18540, 2024
72024
The system can't perform the operation now. Try again later.
Articles 1–20