- add fp32 and flax (6760c4ad746f98dd2bcabe46e41e957fc8c39498) Co-authored-by: Sanchit Gandhi <sanchit-gandhi@users.noreply.huggingface.co>