JAX/Flaxを使ã£ã¦MNISTã‚’å¦ç¿’ã•ã›ã¦ã¿ã‚‹ | TC3æ ªå¼ä¼šç¤¾ï½œGIG INNOVATED.本記事ã¯20分程度ã§ãŠèªã¿ã„ãŸã ã‘ã¾ã™ã€‚ ã“ã‚“ã«ã¡ã¯ã€‚TC3データサイエンス部門ã®æ¢…本ã§ã™ã€‚ 普段ã¯PyTorchを使ã£ã¦ã„ã‚‹ã®ã§ã™ãŒã€æ–°ã—ã„ライブラリを触るã®ã‚‚勉強ã«ãªã‚‹ã¨æ€ã„ã¾ã™ã®ã§ã€ä»Šæ—¥ã¯æ–°é€²æ°—é‹ã®æ·±å±¤å¦ç¿’ライブラリã§ã‚ã‚‹JAX/Flaxを使ã£ã¦ã€MNISTã‚’å¦ç¿’ã•ã›ã¦ã¿ã‚ˆã†ã¨æ€ã„ã¾ã™ã€‚ ã¯ã˜ã‚㫠皆ã•ã‚“ã”å˜çŸ¥ã®é€šã‚Šã€TensorFlowã€Kerasã€PyTorch(Chainer…)ã¨è¿‘å¹´ã¯æ§˜ã€…ãªæ·±å±¤å¦ç¿’ライブラリãŒä½¿ã‚ã‚Œã¦ã„ã¾ã™ã€‚最近ã€JAXã¨ã„ã†ãƒ©ã‚¤ãƒ–ラリãŒè©±é¡Œã«ãªã£ã¦ã„ã‚‹ã‚‚ã®ã®ã€å分ã™ãŽã‚‹ãƒ©ã‚¤ãƒ–ラリãŒã‚ã‚‹ä¸ã§ãªãœJAXãŒæ–°ãŸã«å‡ºã¦ããŸã®ã§ã—ょã†ã‹ï¼Ÿï¼ˆãã—ã¦ãªãœä½¿ã†ã¹ããªã®ã‹ï¼‰ã€‚ã“ã®ç†ç”±ã«ã¯å¾Œç™ºãƒ©ã‚¤ãƒ–ラリã®å¼·ã¿ã¨ã—ã¦ã€å…ˆè¡Œãƒ©ã‚¤ãƒ–ラリã®å•é¡Œç‚¹ã‚’改良ã—ã¦ã„ã‚‹ã¨ã„ã†ç‚¹ãŒæŒ™ã’られã¾ã™ã€‚ç¾çŠ¶ä»¥ä¸‹ã®ã‚ˆã†ãªåˆ©ç‚¹ãŒæŒ™ã’られã¾ã™ XLAコンパイルã«ã‚ˆã‚‹é«˜é€Ÿæ€§ 厳密ãªä¹±æ•°ã®ç®¡ç†ã«ã‚ˆã‚‹å†ç¾æ€§ã®æ‹…ä¿
{{#tags}}- {{label}}
{{/tags}}