Been digging through the HF repo and haven’t been able to locate the training scripts for flax / jax models. Are there scripts for training the various jax model implementations from scratch?
1 Like
Hi there! You can find the official examples at https://github.com/huggingface/transformers/tree/master/examples/flax as well as notebooks at https://github.com/huggingface/notebooks/tree/master/examples.
1 Like