原创 關於要替代TensorFlow的JAX,你知道多少?

這個簡短的教程將介紹關於 JAX 的基礎知識。JAX 是一個 Python 庫,它通過函數轉換來增強 numpy 和 Python 代碼,使運行機器學習程序中常見的操作輕而易舉。具體來說,它會使得編寫標準 Python / numpy 代碼