JAX

2年前发布 4,020 0 0

GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。 JAX的主要功能是包括： grad...

收录时间：
2024-10-18
打开网站手机查看
JAXJAX
JAX打开网站

GoogleJAX是一个用于变换数值函数的机器学习框架，Google称其为为结合了修改版本的Autograd（通过函数微分自动获得梯度函数）和TensorFlow的XLA（加速线性代数）。该框架的设计尽可能遵循NumPy的结构和工作流程，并与TensorFlow和PyTorch等各种现有框架协同工作。

JAX的主要功能是包括：

  • grad：自动微分
  • jit：编译
  • vmap：自动矢量化
  • pmap：SPMD编程

数据统计

相关导航

暂无评论

none
暂无评论...

标签云


Fatal error: Uncaught TypeError: Cannot access offset of type string on string in /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/inc/functions/io-search.php:441 Stack trace: #0 /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/inc/functions/io-footer.php(374): io_search_body_html() #1 /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/inc/functions/io-footer.php(336): io_nav_search_html() #2 /www/wwwroot/ai.itotii.com/wp-includes/class-wp-hook.php(341): io_get_footer_content() #3 /www/wwwroot/ai.itotii.com/wp-includes/class-wp-hook.php(365): WP_Hook->apply_filters() #4 /www/wwwroot/ai.itotii.com/wp-includes/plugin.php(522): WP_Hook->do_action() #5 /www/wwwroot/ai.itotii.com/wp-includes/general-template.php(3224): do_action() #6 /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/footer.php(41): wp_footer() #7 /www/wwwroot/ai.itotii.com/wp-includes/template.php(814): require_once('...') #8 /www/wwwroot/ai.itotii.com/wp-includes/template.php(749): load_template() #9 /www/wwwroot/ai.itotii.com/wp-includes/general-template.php(92): locate_template() #10 /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/single-sites.php(55): get_footer() #11 /www/wwwroot/ai.itotii.com/wp-includes/template-loader.php(132): include('...') #12 /www/wwwroot/ai.itotii.com/wp-blog-header.php(19): require_once('...') #13 /www/wwwroot/ai.itotii.com/index.php(17): require('...') #14 {main} thrown in /www/wwwroot/ai.itotii.com/wp-content/themes/onenav/inc/functions/io-search.php on line 441