FluxML/model-zoo

Please do not feed the models

What it solves

此倉庫作為 Flux 機器學習函式庫的參考實作與示範集合。它為使用者提供即用的起始範例,以建構自己的模型,涵蓋跨不同模態的廣泛機器學習任務。

How it works

此專案以「zoo」形式組織獨立範例。每個模型實作皆置於各自的資料夾,使用 Julia 專案環境來鎖定特定套件版本以確保可重現性。使用者可透過 Julia 的套件管理器 (Pkg) 來建立環境,然後執行模型腳本。

Who it’s for

  • 機器學習實務者:尋找 Julia 中常見架構的參考實作。
  • 開發者:想透過具體範例學習如何使用 Flux 函式庫。
  • 研究人員:需要快速的視覺、文字或強化學習模型起始點。

Highlights

  • 多樣的模型類別:包含 CNN、RNN、VAE、GAN 與 NanoGPT 的實作。
  • GPU 支援:大多數模型設計為 CPU/GPU 無關,並支援 NVIDIA GPU(CUDA)。
  • ** وكذلك **
  • 可重現的環境:每個範例使用本地的 Project 與 Manifest 檔案,確保使用相同的套件版本。
  • Gitpod 整合:提供雲端 IDE 選項,能快速執行範例而無需本機設定。