Support loading and fine-tuning TensorFlow models
Open
Priority: P2 - no schedule
- Dominant language
- Jupyter Notebook
- Stars
- 7.3k
- Forks
- 833
- Avg merge
- 5h 11m
- Merged PRs (30d)
- 5
Description
We should create a guide on how to load TensorFlow models and integrate them into a Flax training workflow for fine-tuning. For now we should avoid creating additional tooling.
References:
* [call_tf](https://github.com/google/jax/blob/main/jax/experimental/jax2tf/call_tf.py#L59)
* [deepmind/tf2jax](https://github.com/deepmind/tf2jax)
Contributor guide
Assessment
This issue has not been assessed yet.