dibyaghosh/pretrained_vision

Utilities for converting pretrained checkpoints to jaxrl_m compatible format

Python