rocm_jax/docs/hero.html

8 lines
398 B
HTML
Raw Permalink Normal View History

<div class="hero">
<div class="hero-left">
<img src="./_static/jax_logo_250px.png"/>
<h2>High performance array computing</h2>
<p>JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning.</p>
</div>
<img src="./_static/jax-hero.svg"/>
</div>