AI開發(fā)框架
JAX
GoogleJAX是一個(gè)用于變換數(shù)值函數(shù)的機(jī)器學(xué)習(xí)框架,Google稱其為為結(jié)合了修改版本的Autograd(通過(guò)函數(shù)微分自動(dòng)獲得梯度函數(shù))和TensorFlow的XLA(加速線性代數(shù))。該框架的設(shè)計(jì)盡可能遵循NumPy的結(jié)構(gòu)和工作流程,并與TensorFlow和PyTorch等各種現(xiàn)有框架協(xié)同工作。
JAX的主要功能是包括:
- grad:自動(dòng)微分
- jit:編譯
- vmap:自動(dòng)矢量化
- pmap:SPMD編程

數(shù)據(jù)評(píng)估
關(guān)于JAX特別聲明
本站OpenI提供的JAX都來(lái)源于網(wǎng)絡(luò),不保證外部鏈接的準(zhǔn)確性和完整性,同時(shí),對(duì)于該外部鏈接的指向,不由OpenI實(shí)際控制,在2023年 5月 9日 上午11:23收錄時(shí),該網(wǎng)頁(yè)上的內(nèi)容,都屬于合規(guī)合法,后期網(wǎng)頁(yè)的內(nèi)容如出現(xiàn)違規(guī),可以直接聯(lián)系網(wǎng)站管理員進(jìn)行刪除,OpenI不承擔(dān)任何責(zé)任。
相關(guān)導(dǎo)航
暫無(wú)評(píng)論...