Cover Image

gfnx: Fast and Scalable Library for Generative Flow Networks in JAX

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Tiapkin, Daniil, Agarkov, Artem, Morozov, Nikita, Maksimov, Ian, Tsyganov, Askar, Gritsaev, Timofei, Samsonov, Sergey
Format:	Preprint
Published:	2025
Subjects:	Machine Learning
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

Optimizing Backward Policies in GFlowNets via Trajectory Likelihood Maximization
by: Gritsaev, Timofei, et al.
Published: (2024)

Learning Shortest Paths with Generative Flow Networks
by: Morozov, Nikita, et al.
Published: (2026)

Revisiting Non-Acyclic GFlowNets in Discrete Environments
by: Morozov, Nikita, et al.
Published: (2025)

Adaptive Destruction Processes for Diffusion Samplers
by: Gritsaev, Timofei, et al.
Published: (2025)

Generative Flow Networks as Entropy-Regularized RL
by: Tiapkin, Daniil, et al.
Published: (2023)

Improving GFlowNets with Monte Carlo Tree Search
by: Morozov, Nikita, et al.
Published: (2024)

XLand-MiniGrid: Scalable Meta-Reinforcement Learning Environments in JAX
by: Nikulin, Alexander, et al.
Published: (2023)

Matrix-Free Two-to-Infinity and One-to-Two Norms Estimation
by: Tsyganov, Askar, et al.
Published: (2025)

Improved High-Probability Bounds for the Temporal Difference Learning Algorithm via Exponential Stability
by: Samsonov, Sergey, et al.
Published: (2023)

Mechanistic Permutability: Match Features Across Layers
by: Balagansky, Nikita, et al.
Published: (2024)

Narrowing the Gap between Adversarial and Stochastic MDPs via Policy Optimization
by: Tiapkin, Daniil, et al.
Published: (2024)

On Global Convergence Rates for Federated Softmax Policy Gradient under Heterogeneous Environments
by: Labbi, Safwan, et al.
Published: (2025)

Refined Analysis of Entropy-Regularized Actor-Critic
by: Labbi, Safwan, et al.
Published: (2026)

Beyond Softmax and Entropy: Convergence Rates of Policy Gradients with f-SoftArgmax Parameterization & Coupled Regularization
by: Labbi, Safwan, et al.
Published: (2026)

Learn Your Reference Model for Real Good Alignment
by: Gorbatovski, Alexey, et al.
Published: (2024)

Scalable Knowledge Editing for Mixture-of-Experts LLMs via Tensor-Structured Updates
by: Maksimov, Roman, et al.
Published: (2026)

Analyze Feature Flow to Enhance Interpretation and Steering in Language Models
by: Laptev, Daniil, et al.
Published: (2025)

Federated UCBVI: Communication-Efficient Federated Regret Minimization with Heterogeneous Agents
by: Labbi, Safwan, et al.
Published: (2024)

DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
by: Rush, Keith, et al.
Published: (2024)

Finite-Sample Convergence Bounds for Trust Region Policy Optimization in Mean-Field Games
by: Ocello, Antonio, et al.
Published: (2025)

jinns: a JAX Library for Physics-Informed Neural Networks
by: Gangloff, Hugo, et al.
Published: (2024)

Rates of convergence for density estimation with generative adversarial networks
by: Puchkin, Nikita, et al.
Published: (2021)

Scalable LinUCB: Low-Rank Design Matrix Updates for Recommenders with Large Action Spaces
by: Shustova, Evgenia, et al.
Published: (2025)

Scalable Mamba-Based Message-Passing Neural Decoder for Error-Correcting Codes
by: Gusev, Rostislav, et al.
Published: (2026)

Statistical analysis of Inverse Entropy-regularized Reinforcement Learning
by: Belomestny, Denis, et al.
Published: (2025)

Pretraining Generative Flow Networks with Inexpensive Rewards for Molecular Graph Generation
by: Pandey, Mohit, et al.
Published: (2025)

Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX
by: Bonnet, Clément, et al.
Published: (2023)

CLAX: Fast and Flexible Neural Click Models in JAX
by: Hager, Philipp, et al.
Published: (2025)

SNNAX -- Spiking Neural Networks in JAX
by: Lohoff, Jamie, et al.
Published: (2024)

Steering LLM Reasoning Through Bias-Only Adaptation
by: Sinii, Viacheslav, et al.
Published: (2025)

BlackJAX: Composable Bayesian inference in JAX
by: Cabezas, Alberto, et al.
Published: (2024)

Convergence, Sticking and Escape: Stochastic Dynamics Near Critical Points in SGD
by: Dudukalov, Dmitry, et al.
Published: (2025)

Jasmine: A Simple, Performant and Scalable JAX-based World Modeling Codebase
by: Mahajan, Mihir, et al.
Published: (2025)

Proximal Point Nash Learning from Human Feedback
by: Tiapkin, Daniil, et al.
Published: (2025)

Demonstration-Regularized RL
by: Tiapkin, Daniil, et al.
Published: (2023)

BindGPT: A Scalable Framework for 3D Molecular Design via Language Modeling and Reinforcement Learning
by: Zholus, Artem, et al.
Published: (2024)

Midpoint Generative Models
by: Shlenskii, Daniil, et al.
Published: (2026)

EEG-Reptile: An Automatized Reptile-Based Meta-Learning Library for BCIs
by: Berdyshev, Daniil A., et al.
Published: (2024)

Gaussian Approximation for Asynchronous Q-learning
by: Rubtsov, Artemy, et al.
Published: (2026)

ESSA: Evolutionary Strategies for Scalable Alignment
by: Korotyshova, Daria, et al.
Published: (2025)