Moksh Jain
Moksh Jain
Mila, Université de Montréal
Verified email at - Homepage
Cited by
Cited by
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
DROCC: Deep robust one-class classification
S Goyal, A Raghunathan, M Jain, HV Simhadri, P Jain
International Conference on Machine Learning, 3711-3721, 2020
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
Deup: Direct epistemic uncertainty prediction
S Lahlou*, M Jain*, H Nekoei, VI Butoi, P Bertin, J Rector-Brooks, ...
Transactions on Machine Learning Research, 2023
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
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
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
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
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
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
Stochastic Generative Flow Networks
L Pan, D Zhang, M Jain, L Huang, Y Bengio
arXiv preprint arXiv:2302.09465, 2023
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
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
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
Proximal Policy Optimization for Improved Convergence in IRGAN
M Jain
SGO & ML, NeurIPS 2019, 2019
Amortizing intractable inference in large language models
EJ Hu, M Jain, E Elmoznino, Y Kaddar, G Lajoie, Y Bengio, N Malkin
arXiv preprint arXiv:2310.04363, 2023
BatchGFN: Generative Flow Networks for Batch Active Learning
SA Malik, S Lahlou, A Jesson, M Jain, N Malkin, T Deleu, Y Bengio, Y Gal
arXiv preprint arXiv:2306.15058, 2023
Multi-Fidelity Active Learning with GFlowNets
A Hernandez-Garcia, N Saxena, M Jain, CH Liu, Y Bengio
arXiv preprint arXiv:2306.11715, 2023
A community effort in SARS‐CoV‐2 drug discovery
J Schimunek, P Seidl, K Elez, T Hempel, T Le, F Noé, S Olsson, L Raich, ...
Molecular Informatics 43 (1), e202300262, 2024
Pre-Training and Fine-Tuning Generative Flow Networks
L Pan, M Jain, K Madan, Y Bengio
arXiv preprint arXiv:2310.03419, 2023
The system can't perform the operation now. Try again later.
Articles 1–20