純 Crystal 神經網路學習庫 SHAInet

開發工具

SHAInet 是以純 Crystal 語言編寫的機器學習庫,支持 CPU/GPU 加速,可用於訓練和運行神經網路,並能導入 HuggingFace 模型。

此 GitHub 開源專案 SHAInet 是一個用純 Crystal 語言實現的機器學習庫,專注於神經網路的訓練與推理。README 說明,它最初為生物啟發神經網路研究設計,現已發展成通用庫,支持多種層類型、激活函數及訓練算法如 SGD、Adam 等。特色功能包括 CUDA GPU 加速、流數據處理大型數據集,以及無需 Python 即可直接導入 HuggingFace 模型的 SafeTensors 格式,支持 F16、BF16 等多種精度。此外,它支援 Transformer 和現代 NLP 應用,適用於 Crystal 開發者進行 AI 研究、模型訓練或部署。專案提供詳細的安裝與使用範例,讓使用者能快速上手進行 XOR 分類、Iris 資料集測試等任務。

Stars197
Forks19
語言Crystal
分類開發工具
標籤
crystalmachine-learningneural-networkdeep-learningcudahuggingfacetransformers
GitHub Topics
convolutional-neural-networkscrystaldeep-learningdeep-neural-networksmachine-learningneural-network