lucidrains/alphafold3-pytorch

何を解決するか

このプロジェクトは、生体分子相互作用の正確な構造予測を目的としたAlphaFold 3のPyTorch実装を提供します。研究者が複雑な生物分子（タンパク質やその他の生体分子を含む）における原子の3次元座標を予測できるようにします。これは生物学的機能の理解や創薬に不可欠です。

動作方法

AlphaFold 3論文で説明されたアルゴリズムをPyTorchコードに変換しています。原子位置のサンプリングに拡散ベースのアプローチを採用し、以下の専門モジュールを統合しています：

• 入力処理：PDB（Protein Data Bank）データ（MSAやテンプレートを含む）を処理します。
• Pairformerスタック：構造情報処理のコアアーキテクチャコンポーネントです。
• 拡散モジュール：原子エンコーダ、トークントランスフォーマー、原子デコーダが連携して原子の3次元座標を精緻化・生成します。
• 学習パイプライン：mmCIFファイルから高品質な学習データを準備するための、PDBデータセットのフィルタリングおよびクラスタリングスクリプトを含みます。

対象ユーザー

• 計算生物学者：生体分子複合体の構造予測が必要な方。
• 機械学習研究者：PyTorch環境でAlphaFold 3のアーキテクチャを実装または研究したい方。
• バイオインフォマティクスエンジニア：構造生物学モデル用に大規模なPDBデータセットを準備・キュレーションするツールが必要な方。

特徴

• 完全なPyTorch実装：AlphaFold 3アーキテクチャをPyTorchエコシステムに導入し、カスタマイズや学習を容易にします。
• 包括的なデータパイプライン：RCSB Protein Data BankからPDBデータをダウンロード・フィルタリング・クラスタリングする詳細スクリプトを含みます。
• エンドツーエンドワークフロー：PDBファイルの解析からサンプリングされた原子位置の生成までをサポートします。
• Docker統合：GPU対応の一貫した環境セットアップを可能にするDockerfileを提供します。