{"html":"<iframe src=\"https://hatenablog-parts.com/embed?url=https%3A%2F%2Frandommemory.hatenablog.com%2Fentry%2F2022%2F04%2F07%2F212757\" title=\"JAX - \u3089\u3093\u3060\u3080\u306a\u8a18\u61b6\" class=\"embed-card embed-blogcard\" scrolling=\"no\" frameborder=\"0\" style=\"display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;\"></iframe>","type":"rich","image_url":null,"blog_url":"https://randommemory.hatenablog.com/","author_name":"derwind","description":"\u81ea\u52d5\u5fae\u5206\u306b\u7279\u5316\u3057\u305f NumPy \u3068\u304b\u8a00\u308f\u308c\u308b JAX\u3002https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/quickstart.ipynb \u3068\u3057\u3066 Colab \u7528\u306e\u30b5\u30f3\u30d7\u30eb\u304c\u3042\u3063\u305f\u3002 size = 3000 x = random.normal(key, (size, size), dtype=jnp.float32) %timeit jnp.dot(x, x.T).block_until_ready() # runs on the GPU The slowest run took 55.35 \u2026","author_url":"https://blog.hatena.ne.jp/derwind/","title":"JAX","blog_title":"\u3089\u3093\u3060\u3080\u306a\u8a18\u61b6","categories":["machine_learning"],"provider_url":"https://hatena.blog","url":"https://randommemory.hatenablog.com/entry/2022/04/07/212757","published":"2022-04-07 21:27:57","provider_name":"Hatena Blog","height":"190","width":"100%","version":"1.0"}