English
Related papers

Related papers: torchgfn: A PyTorch GFlowNet library

200 papers

The rnn package provides components for implementing a wide range of Recurrent Neural Networks. It is built withing the framework of the Torch distribution for use with the nn package. The components have evolved from 3 iterations, each…

Neural and Evolutionary Computing · Computer Science 2015-12-18 Nicholas Léonard , Sagar Waghmare , Yang Wang , Jin-Hwa Kim

There are many frameworks for deep generative modeling, each often presented with their own specific training algorithms and inference methods. Here, we demonstrate the connections between existing deep generative models and the recently…

Machine Learning · Computer Science 2023-02-01 Dinghuai Zhang , Ricky T. Q. Chen , Nikolay Malkin , Yoshua Bengio

The Generative Flow Network is a probabilistic framework where an agent learns a stochastic policy for object generation, such that the probability of generating an object is proportional to a given reward function. Its effectiveness has…

Machine Learning · Computer Science 2022-10-10 Ling Pan , Dinghuai Zhang , Aaron Courville , Longbo Huang , Yoshua Bengio

As the computational requirements for machine learning systems and the size and complexity of machine learning frameworks increases, essential framework innovation has become challenging. While computational needs have driven recent…

Access to vast amounts of data along with affordable computational power stimulated the reincarnation of neural networks. The progress could not be achieved without adequate software tools, lowering the entry bar for the next generations of…

Machine Learning · Computer Science 2019-10-22 Tomasz Kornuta

We introduce GPflux, a Python library for Bayesian deep learning with a strong emphasis on deep Gaussian processes (DGPs). Implementing DGPs is a challenging endeavour due to the various mathematical subtleties that arise when dealing with…

Generative recommendations (GR), which usually include item tokenizers and generative Large Language Models (LLMs), have demonstrated remarkable success across a wide range of scenarios. The majority of existing research efforts primarily…

Information Retrieval · Computer Science 2025-11-25 Yejing Wang , Shengyu Zhou , Jinyu Lu , Qidong Liu , Xinhang Li , Wenlin Zhang , Feng Li , Pengjie Wang , Jian Xu , Bo Zheng , Xiangyu Zhao

Generative Flow Networks (GFlowNets) are a class of generative models that sample objects in proportion to a specified reward function through a learned policy. They can be trained either on-policy or off-policy, needing a balance between…

Machine Learning · Computer Science 2025-03-04 Dominic Phillips , Flaviu Cipcigan

Many score-based active learning methods have been successfully applied to graph-structured data, aiming to reduce the number of labels and achieve better performance of graph neural networks based on predefined score functions. However,…

Machine Learning · Computer Science 2023-04-25 Yinchuan Li , Zhigang Li , Wenqian Li , Yunfeng Shao , Yan Zheng , Jianye Hao

We introduce tf_geometric, an efficient and friendly library for graph deep learning, which is compatible with both TensorFlow 1.x and 2.x. tf_geometric provides kernel libraries for building Graph Neural Networks (GNNs) as well as…

Machine Learning · Computer Science 2021-01-28 Jun Hu , Shengsheng Qian , Quan Fang , Youze Wang , Quan Zhao , Huaiwen Zhang , Changsheng Xu

Generative Flow Networks (GFlowNets) excel at sampling diverse, high-reward objects. In many practical applications where active reward queries are infeasible, these models must be trained using static offline datasets. Prevailing training…

Machine Learning · Computer Science 2026-05-26 Ruishuo Chen , Xun Wang , Rui Hu , Zhuoran Li , Longbo Huang

We introduce BatchGFN -- a novel approach for pool-based active learning that uses generative flow networks to sample sets of data points proportional to a batch reward. With an appropriate reward function to quantify the utility of…

Machine Learning · Computer Science 2023-06-28 Shreshth A. Malik , Salem Lahlou , Andrew Jesson , Moksh Jain , Nikolay Malkin , Tristan Deleu , Yoshua Bengio , Yarin Gal

Generative Flow Networks (GFlowNets) are powerful samplers for compositional objects that, by design, sample proportionally to a given non-negative reward. Nonetheless, in practice, they often struggle to explore the reward landscape…

Machine Learning · Computer Science 2026-03-17 Pedro Dall'Antonia , Tiago da Silva , Daniel Augusto de Souza , César Lincoln C. Mattos , Diego Mesquita

Software systems are increasingly relying on deep learning components, due to their remarkable capability of identifying complex data patterns and powering intelligent behaviour. A core enabler of this change in software development is the…

Machine learning has been revolutionizing our world over the last few years and is also increasingly exploited in several areas of physics, including quantum dynamics and control.The need for a framework that brings together machine…

Quantum Physics · Physics 2025-01-31 Dimitris Koutromanos , Dionisis Stefanatos , Emmanuel Paspalakis

Generative flow networks utilize a flow-matching loss to learn a stochastic policy for generating objects from a sequence of actions, such that the probability of generating a pattern can be proportional to the corresponding given reward.…

Machine Learning · Computer Science 2025-09-26 Leo Maxime Brunswic , Haozhi Wang , Shuang Luo , Jianye Hao , Amir Rasouli , Yinchuan Li

Generative Flow Networks (GFlowNets; GFNs) are a family of energy-based generative methods for combinatorial objects, capable of generating diverse and high-utility samples. However, consistently biasing GFNs towards producing high-utility…

Machine Learning · Computer Science 2024-11-04 Elaine Lau , Stephen Zhewen Lu , Ling Pan , Doina Precup , Emmanuel Bengio

Generative Flow Networks (GFlowNets) have recently emerged as a suitable framework for generating diverse and high-quality molecular structures by learning from rewards treated as unnormalized distributions. Previous works in this framework…

Machine Learning · Computer Science 2025-06-13 Mohit Pandey , Gopeshh Subbaraj , Artem Cherkasov , Martin Ester , Emmanuel Bengio

In order to contrast the explosion in size of state-of-the-art machine learning models that can be attributed to the empirical advantages of over-parametrization, and due to the necessity of deploying fast, sustainable, and private…

Machine Learning · Computer Science 2020-04-30 Michela Paganini , Jessica Forde

There is a need for open-source libraries in emission tomography that (i) use modern and popular backend code to encourage community contributions and (ii) offer support for the multitude of reconstruction techniques available in recent…