Skip to content

Artifex

Protein Extensions API

Initializing search

Artifex

Home
Getting Started
Getting Started
User Guide
User Guide
- Concepts
  Concepts
- Models
  Models
- Training
  Training
- Data
  Data
  - Data Loading Overview
  - Data Loading Guide
- Modalities
  Modalities
- Inference
  Inference
- Advanced
  Advanced
- Integrations
  Integrations
API Reference
API Reference
- Core
  Core
- Models
  Models
  - VAE
  - GAN
  - Diffusion
  - Flow
  - Autoregressive
  - Energy-Based
- Training
  Training
  - Trainer
  - Model-Specific Trainers
    Model-Specific Trainers
    
    VAE Trainer
    
    GAN Trainer
    
    Diffusion Trainer
    
    Flow Trainer
  - RL Trainers
    RL Trainers
    
    REINFORCE
    
    PPO
    
    GRPO
    
    DPO
  - Callbacks
    Callbacks
    
    Base Classes
    
    Logging
    
    Profiling
    
    Checkpointing
    
    Early Stopping
  - Distributed
    Distributed
    
    Device Mesh
    
    Data Parallel
    
    Device Placement
    
    Distributed Metrics
    
    Model Parallel
- Data
  Data
  - Data Loaders
Examples
Examples
- Overview
- Diffusion Models
  Diffusion Models
  - Standalone Simple Diffusion
  - DiT Demo
- Energy-Based Models
  Energy-Based Models
  - Simple EBM
- Basic Tutorials
  Basic Tutorials
- Advanced Tutorials
  Advanced Tutorials
  - Advanced Training
  - Advanced VAE
- VAE Models
  VAE Models
  - Multi-Beta VAE Benchmark
- Protein Modeling
  Protein Modeling
- Exploratory Workflows
  Exploratory Workflows
  - Exploratory Advanced GAN
  - Exploratory Protein Diffusion
- Validation Utilities
  Validation Utilities
  - Validation Protein Tech Check
- Geometric Models
  Geometric Models
- Audio Generation
  Audio Generation
  - Standalone Audio Generation
- Text Generation
  Text Generation
  - Standalone Text Generation
- Multimodal Models
  Multimodal Models
  - Standalone Image-Text
- Framework Features
  Framework Features
  - Framework Features Demo
- Loss Functions
  Loss Functions
  - Loss Examples
- Sampling Methods
  Sampling Methods
- All Examples
Generative Models
Generative Models
Training
Training
Data & Modalities
Data & Modalities
Inference & Generation
Inference & Generation
Evaluation
Evaluation
Guides
Guides
Advanced Topics
Advanced Topics
- Extensions
- Loss Functions
Development
Development
Community
Community
- Contributing
- FAQ
Reference Library
Reference Library
- API Companion Pages
  API Companion Pages
  - Protein Extension Configuration
  - Configuration API
  - Protein Dataset API
  - CrossDocked Dataset API
  - Extensions API
  - Protein Extensions API Protein Extensions API
    On this page
    
    Overview
    
    Related Documentation
  - Factory API
  - Protein Point Cloud Model API
- Architecture
  Architecture
  - Extensions Architecture
- Retained Benchmark Reference
  Retained Benchmark Reference
- CLI Reference
  CLI Reference
  - CLI Entrypoint
  - Benchmark
  - Config
  - Convert
  - Evaluate
  - Generate
  - Serve
  - Train
- Config Reference
  Config Reference
- Core Reference
  Core Reference
  - Adversarial
  - Ancestral
  - Architecture Overview
  - Base
  - Benchmarks
  - BlackJAX Samplers
  - Causal
  - Checkpointing
  - Config Commands
  - Configuration
  - Continuous
  - Distribution Base
  - Device Manager
  - Device Testing
  - Diffusion
  - Discrete
  - Distance
  - Divergence
  - Evaluation
  - FID
  - Flash Attention
  - GAN Configuration
  - Geometric
  - Inception Score
  - Core
  - Information
  - Jax Config
  - Logging
  - Mcmc
  - Metric Ops
  - Config Migration
  - Mixture
  - Ode
  - Perceptual
  - Performance
  - Perplexity
  - Pipeline
  - Positional
  - Precision Recall
  - Quality
  - Reconstruction
  - Registry
  - Regularization
  - Residual
  - Resnet
  - Runner
  - Sde
  - Statistical
  - Templates
  - Transformations
  - Transformers
  - Unified Configuration Backend
  - Configuration Validation Helpers
  - Versioning
- Data Reference
  Data Reference
  - Audio
  - Base
  - Bookcorpus
  - Bpe
  - Character
  - Cifar
  - Coco
  - Collators
  - Custom Audio
  - Custom Image
  - Custom Multimodal
  - Custom Text
  - Custom Video
  - Protein Dataset Module
  - Ffhq
  - Image
  - Imagenet
  - Data Package
  - Librispeech
  - Pipeline
  - Protein Dataset
  - Registry
  - Remote
  - Samplers
  - Sentencepiece
  - Structured
  - Text
  - Tfrecord
  - Ucf101
  - Vctk
  - Video
  - Webdataset
  - Wikipedia
  - Word
- Extensions Owner Pages
  Extensions Owner Pages
  - Augmentation
  - Backbone
  - Constraints
  - Embeddings
  - Extensions
  - Features
  - Mixin
  - Registry
  - Spectral
  - Temporal
  - Tokenization
  - Utils
- Factory Reference
  Factory Reference
  - Autoregressive
  - Core
  - Diffusion
  - Ebm
  - Flow
  - Gan
  - Geometric
  - Vae
- Generative Models Reference
  Generative Models Reference
  - Generative Models
- Guide Addenda
  Guide Addenda
  - Protein-Ligand Benchmark Guide
  - Molecular Modality Guide
- Inference Reference
  Inference Reference
  - Current Inference Pages
    Current Inference Pages
    
    Inference Reference
    
    Production
  - Coming Soon
    Coming Soon
    
    Adapter
    
    Ancestral
    
    Autoregressive Generator
    
    Base
    
    Beam Search
    
    Caching
    
    Classifier Free
    
    Compilation
    
    Diffusion Generator
    
    Distillation
    
    Dynamic
    
    Energy Generator
    
    Flow Generator
    
    Gan Generator
    
    Grpc
    
    Latency
    
    Lora
    
    Memory
    
    Middleware
    
    Nucleus
    
    Onnx
    
    Padding
    
    Pipeline
    
    Prefix Tuning
    
    Prompt Tuning
    
    Pruning
    
    Quantization
    
    Rest
    
    Stateless
    
    Streaming
    
    Temperature
    
    Tensorrt
    
    Tfjs
    
    Throughput
    
    Top K
    
    Vae Generator
- Modalities Reference
  Modalities Reference
- Model Reference
  Model Reference
  - Base
  - Beta Vae
  - Conditional
  - CycleGAN
  - Dcgan
  - Ddim
  - Ddpm
  - Decoders
  - Audio Diffusion
  - DiT
  - Ebm
  - Encoders
  - Glow
  - Graph
  - Guidance
  - Iaf
  - Latent
  - Lsgan
  - Made
  - Maf
  - Mcmc
  - Mesh
  - Neural Spline
  - Patchgan
  - Pixel Cnn
  - Point Cloud
  - Protein Graph
  - Protein Point Cloud
  - Real Nvp
  - Score
  - SE3 Molecular Flow
  - Stable Diffusion
  - Stylegan3
  - Transformer
  - Voxel
  - Vq Vae
  - Wavenet
  - WGAN
- Coming Soon
  Coming Soon
- Notebooks
  Notebooks
  - Notebooks
- Notebook Source Views
  Notebook Source Views
  - None
  - None
- Papers
  Papers
  - Artifex: A Generative Modeling Research Library Under Active Development
- Roadmap
  Roadmap
  - Planned Modules Roadmap
  - Planned Example Topics
- Scaling Reference
  Scaling Reference
  - Mesh Utils
  - Sharding
- Theory
  Theory
  - Molecular Constraints Theory
- Training Reference
  Training Reference
  - Current Training Pages
    Current Training Pages
    
    Training Systems
    
    Base Classes
    
    Checkpointing
    
    Early Stopping
    
    Logging
    
    Profiling
    
    Training Factories
    
    Gradient Accumulation
    
    Data Parallel
    
    Device Placement
    
    Distributed Metrics
    
    Device Mesh
    
    Training Utils
    
    VAE Trainer
    
    GAN Trainer
    
    Diffusion Trainer
    
    Flow Trainer
    
    Energy Trainer
    
    Autoregressive Trainer
    
    REINFORCE
    
    PPO
    
    GRPO
    
    DPO
  - Coming Soon
    Coming Soon
    
    Adamw
    
    Adafactor
    
    Lion
    
    Scheduler
    
    Optax Wrappers
    
    Exponential
    
    Linear
    
    Cosine
    
    Mixed Precision
    
    Tracking
    
    Visualization
    
    Model Parallel
- Tutorials
  Tutorials
  - Protein Modeling Tutorial
- Utilities Reference
  Utilities Reference
  - Current Utility Pages
    Current Utility Pages
    
    Dependency Analyzer
    
    Device Utilities
    
    File Utils
    
    Logger
    
    Metrics
    
    MLflow
    
    Protein Visualization Compatibility
    
    W&B
  - Coming Soon
    Coming Soon
    
    Attention Vis
    
    Color
    
    Dtype
    
    Env
    
    File
    
    Flax Utils
    
    Formats
    
    Image Grid
    
    Latent Space
    
    Math
    
    Memory
    
    Performance
    
    Plotting
    
    Postprocessing
    
    Prng
    
    Processing
    
    Registry
    
    Serialization
    
    Shapes
    
    Stability
    
    Stats
    
    Timer
    
    Transforms
    
    Types
    
    Xprof
- Visualization Reference
  Visualization Reference
  - Protein Viz

On this page

Overview
Related Documentation

Home
Reference Library
API Companion Pages

Protein Extensions API¤

Overview¤

API reference for protein modeling extensions.

Related Documentation¤

Protein Modeling Guide
Protein Extensions Guide

Made with Material for MkDocs