JAX implementation of current algos

tinkoff-ai / CORL

High-quality single-file implementations of SOTA Offline and Offline-to-Online RL algorithms: AWAC, BC, CQL, DT, EDAC, IQL, SAC-N, TD3+BC, LB-SAC, SPOT, Cal-QL, ReBRAC

https://arxiv.org/abs/2210.07105

Apache License 2.0

1.06k stars 124 forks source link

JAX implementation of current algos #65

Closed chkda closed 1 year ago

chkda commented 1 year ago

By any chance is there a plan on having single file implementation using JAX

vkurenkov commented 1 year ago

Hi,

There are no such plans in the short-term, but feel free to contribute, we can take a look and consider adding quality implementations in.