PrxteinMPNN Documentation

Run on Colab

Welcome to PrxteinMPNN’s documentation! This project provides a functional interface for ProteinMPNN, leveraging the JAX ecosystem for accelerated computation.

Contents:

• API Reference
  • JacobianSpecification
  • RunSpecification
  • SamplingSpecification
  • ScoringSpecification
  • categorical_jacobian()
  • sample()
  • score()
  • io: Input/Output for Structure and Trajectories
  • mpnn: Fetch model parameters
  • model: MPNN model architecture
  • sampling: Sampling methods for MPNN
  • scoring: Scoring functions for MPNN
  • utils: Utility functions for MPNN
• Tutorials
  • Quick Start
  • Installation Guide
  • Basic Usage
  • Getting Started
  • Advanced Topics
• Examples
  • Advanced Example
  • Basic Examples
  • Advanced Examples
  • Code Examples

Key Features

• Increased Transparency: Clear and functional interface for ProteinMPNN

• JAX Compatibility: Efficient computation with JAX’s functional programming paradigm

• Modular Design: Easy updates and extensions

• Performance Optimization: Utilizes JAX’s JIT compilation and vectorization

Installation

pip install -e .

For development:

pip install -e ".[dev]"

Use notes