使用 Flax 的存放庫#
以下程式碼檔案使用 Flax 並提供訓練架構以及豐富的範例。在許多情況下，您也可以找到預先訓練的權重
🤗 Hugging Face#
🤗 Hugging Face 是一個用於建置、訓練和部署最先進機器學習模型的非常受歡迎的函式庫。這些模型可應用於文字、影像和音訊。在組織了 JAX/Flax 社群週 之後，他們現在在他們的儲存庫中擁有超過 5,000 個 Flax/JAX 模型。
🥑 DALLE Mini#
🥑 DALLE Mini 是一個使用 JAX/Flax 實作的文字轉影像的 Transformer 模型，依循 OpenAI 原始 DALLE 論文的想法。
Scenic#
Scenic 是電腦視覺研究及其他領域的程式碼庫/函式庫。Scenic 的主要重點是圍繞基於注意力的模型。Scenic 已成功用於開發多個模態的分類、分割和偵測模型，包括影像、影片、音訊和它們的多模態組合。
Big Vision#
Big Vision 是專為使用雲端 TPU VM 或 GPU 機器訓練大規模視覺模型而設計的程式碼庫。它基於 Jax/Flax 函式庫，並使用 tf.data 和 TensorFlow 資料集以用於可擴充性和可再現的輸入管道。這是 ViT、MLP-Mixer、LiT、UViM 等許多模型的原始程式碼庫。