コンテンツへスキップ ナビゲーションに移動

まったり勉強ノート

JAX

  1. HOME
  2. JAX
11月 17, 2021 / 最終更新日時 : 11月 17, 2021 Shuji Suzuki (shu) プログラミング

JAXとPyTorch、どっちが速いのか検証してみた

高速化が趣味&仕事なので、最近よく目にするJAXの速度が気になってました。このため、今回は日ごろ使っているPyTorchと比較したので、その結果のまとめを紹介します。 結論 結果だけ知りたい方が多いだろうと思ったので先に […]

11月 8, 2021 / 最終更新日時 : 11月 8, 2021 Shuji Suzuki (shu) プログラミング

JAXによる微分可能Smith Watermanアルゴリズムのパフォーマンス測定

最近微分可能な Smith Waterman アルゴリズムというものとJAXのコードが公開されました。今回はこれらを参考に、JAXの勉強がてら何パターンかSmith Watermanアルゴリズムを実装して測定してみたので […]

Shuji Suzuki (shu)

shuji-suzuki

バイオインフォとHPCっぽいことを研究してる企業研究者。主に調べたり勉強したりしたことのまとめを書いてます。

最近の投稿

PyTorch Geometricを使ってVariational Graph Auto-Encodersを作って学習してみる
5月 23, 2022
UniProtのWeb APIを使ってほしい遺伝子のタンパク質配列を取ってくる
5月 7, 2022
ReactomeからPathwayの階層構造とPathwayに関連するGeneのデータを取得する
5月 6, 2022
[勉強ノート] 「線形計算の数理」 7.7.1 Arnoldi法
4月 4, 2022
[勉強ノート] 「線形計算の数理」7.4 QR法
3月 29, 2022
[勉強ノート] 「線形計算の数理」7.3 同時反復法
3月 28, 2022
[勉強ノート] 「線形計算の数理」 7.2.1 べき乗法 基本形
3月 24, 2022
[勉強ノート]「ベイズ推論による機械学習入門」5.7 ニューラルネットワークで紹介されたものをPyTorchで実装してみた
2月 28, 2022
[勉強ノート]「ベイズ推論による機械学習入門」5.6ロジスティック回帰
2月 24, 2022
Optuna V3開発スプリント #2 に行ってきた
12月 14, 2021

最近の投稿

PyTorch Geometricを使ってVariational Graph Auto-Encodersを作って学習してみる

5月 23, 2022

UniProtのWeb APIを使ってほしい遺伝子のタンパク質配列を取ってくる

5月 7, 2022

ReactomeからPathwayの階層構造とPathwayに関連するGeneのデータを取得する

5月 6, 2022

[勉強ノート] 「線形計算の数理」 7.7.1 Arnoldi法

4月 4, 2022

[勉強ノート] 「線形計算の数理」7.4 QR法

3月 29, 2022

[勉強ノート] 「線形計算の数理」7.3 同時反復法

3月 28, 2022

[勉強ノート] 「線形計算の数理」 7.2.1 べき乗法 基本形

3月 24, 2022

[勉強ノート]「ベイズ推論による機械学習入門」5.7 ニューラルネットワークで紹介されたものをPyTorchで実装してみた

2月 28, 2022

[勉強ノート]「ベイズ推論による機械学習入門」5.6ロジスティック回帰

2月 24, 2022

Optuna V3開発スプリント #2 に行ってきた

12月 14, 2021

カテゴリー

  • プログラミング
  • 書評
  • 未分類

アーカイブ

  • 2022年5月
  • 2022年4月
  • 2022年3月
  • 2022年2月
  • 2021年12月
  • 2021年11月
  • 2021年10月
  • 2021年5月
  • 2021年4月
  • 2021年3月
  • 2021年2月

アーカイブ

  • 2022年5月
  • 2022年4月
  • 2022年3月
  • 2022年2月
  • 2021年12月
  • 2021年11月
  • 2021年10月
  • 2021年5月
  • 2021年4月
  • 2021年3月
  • 2021年2月

カテゴリー

  • プログラミング
  • 書評
  • 未分類

Copyright © まったり勉強ノート All Rights Reserved.

Powered by WordPress with Lightning Theme & VK All in One Expansion Unit by Vektor,Inc. technology.

MENU
PAGE TOP Secured By miniOrange