sde

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

Diffusion Models in Medical Imaging (Published in Medical Image Analysis Journal)

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code