versae's picture
Improved version of conversion script Flax → PyTorch
346a10a