Integrating deep learning into first-principles calculations
Modular, extensible bridge between first-principles calculations and the DeepH method
DeepH-dock is a modular, extensible interface platform for quantum materials calculations, dedicated to building efficient and reliable bridges between first-principles calculations and the DeepH (deep learning Hamiltonian) method. This platform integrates multiple density functional theory (DFT) software interfaces, supports DeepH predictions, and provides standardized data processing. It also functions independently as a post-processing tool for DFT calculations.
At the core of DeepH-dock is a unified and flexible interface layer that seamlessly connects mainstream DFT packages with the DeepH workflow, enabling users to generate and utilize deep learning-based Hamiltonians with minimal effort. DeepH-dock offers first-class support for heterogeneous computational environments, allowing researchers to orchestrate complex multi-software workflows through a consistent Python API.
DeepH-dock also establishes a unified data format tailored for machine learning in materials science, facilitating efficient implementations of both force fields and electronic structure methods.
For the most comprehensive usage documentation, please visit https://docs.deeph-pack.com/deeph-dock.
For the latest version-specific documentation of DeepH-pack, please refer to the corresponding repository.
Core Features
-
Multi-Software Compatibility: Seamlessly bridges major DFT packages (e.g., SIESTA, QE, OpenMX, FHI-aims, ABACUS) with the DeepH deep learning family and downstream tight-binding toolchains.
-
High-Performance Computational Core: Implements high-performance algorithms for automated structure generation, overlap matrix calculation, Hamiltonian symmetrization, and fast diagonalization via KPM/Lanczos.
-
Standardized Workflows: Streamlines the full research lifecycle with automated pipelines for DFT data generation, preprocessing, post-analysis, and custom composite workflows.
-
Utility Toolkit: Empowers developers with a robust set of tools for multi-level parallel computing (MPI/Loky), versatile format conversion, and rigorous data integrity validation.
Quick Start
Installation
Publish version:
pip install deepx-dock
Development version:
pip install git+https://github.com/kYangLi/DeepH-dock
Basic Usage
run dock -h,
Usage: dock [OPTIONS] COMMAND [ARGS]...
DeepH-dock: Materials Computation and Data Analysis Toolkit.
Options:
--version Show the version and exit.
-h, --help Show this message and exit.
Commands:
analyze
compute
convert
design
ls List all available commands.
Citation
Any and all use of this software, in whole or in part, should clearly acknowledge and link to this repository.
If you use this code in your academic work, please cite the complete package featuring the latest implementation, methodology, and workflow of DeepH:
@article{li2026deeph,
title={DeepH-pack: A general-purpose neural network package for deep-learning electronic structure calculations},
author={Li, Yang and Wang, Yanzhen and Zhao, Boheng and Gong, Xiaoxun and Wang, Yuxiang and Tang, Zechen and Wang, Zixu and Yuan, Zilong and Li, Jialin and Sun, Minghui and Chen, Zezhou and Tao, Honggeng and Wu, Baochun and Yu, Yuhang and Li, He and da Jornada, Felipe H. and Duan, Wenhui and Xu, Yong },
journal={arXiv preprint arXiv:2601.02938},
year={2026}
}
Application Scenarios
- High-Throughput Materials Calculations: Automated generation and processing of electronic structure data for large material sets
- Deep Learning Training Data Preparation: Preparing standardized training datasets for neural network models
- Tight-Binding Parameter Fitting: Connecting DFT calculations with tight-binding model parameterization
- Method Development and Validation: Rapid implementation and testing of new computational algorithms
Contributing
We welcome contributions from the community! To ensure a consistent and maintainable codebase, please follow the structured workflow below when adding new functionality.
Development Workflow for Contributors
A brief summary is provided below. Complete documentation can be found in the Development Guide.
-
Identify the Target Module
First, determine which major module your contribution belongs to, based on the project architecture (e.g.,
analyze,compute,convert,design). -
Create Your Submodule
Create a new folder for your submodule within the corresponding parent module. For example, if you are implementing an interface between FHI-aims and DeepH, create it under
deepx_dock/convert/fhi_aims/. -
Implement Your Functionality
Package the core logic you wish to add as a well-defined Python class or function. This makes the code reusable and easier to test.
-
Register a Command-Line Interface (CLI)
This is the most important step. To expose your functionality through the main
dockcommand, you must register your function inside the_cli.pyfile within your submodule directory.In this file, define your CLI command using the
clicklibrary and the provided@registerdecorator.
Example:
If you create deepx_dock/convert/fhi_aims/_cli.py with the following content:
import click
from pathlib import Path
from deepx_dock._cli.registry import register
@register(
cli_name="my-func",
cli_help="A brief description of what my-func does.",
cli_args=[
click.argument('fhiaims_dir', type=click.Path()),
click.argument('deeph_dir', type=click.Path()),
click.option(
'--parallel-num', '-p', type=int, default=-1,
help="The parallel processing number, -1 for using all of the cores."
),
],
)
def my_func(fhiaims_dir: Path, deeph_dir: Path, parallel_num: int):
# Your implementation logic here
from .my_wanderful_function import my_real_func
my_real_func(fhiaims_dir, deeph_dir, parallel_num)
Then, on the user end, a new command will be available:
dock convert fhi-aims my-func
The command's usage (arguments, options, help text) is entirely defined by the click decorators specified in cli_args. Please refer to the official click documentation for details on defining arguments and options.
License
This project is licensed under the GPL 3.0 License - see the LICENSE file for details.
Support & Contact
- 📖 Documentation: Documentation Link
- 🐛 Issue Reporting: GitHub Issues
DeepH-dock is community-driven development for quantum materials calculations, aiming to promote openness and reproducibility in computational materials science research.
Metadata
Release files for deepx-dock 0.9.11
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Actions Reset |
|---|---|---|---|---|
| deepx_dock-0.9.11-py3-none-any.whl | Python 3 | none | any | Details for deepx_dock-0.9.11-py3-none-any.whl |
Release files / deepx_dock-0.9.11-py3-none-any.whl
| Download URL | deepx_dock-0.9.11-py3-none-any.whl |
|---|---|
| Size | 194.7 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d6ab45b0b0eeec6520aece7dd3674cc9900d21da93874501b2bcf04d9d4293dd
|
|
BLAKE2b-256 checksum How to use checksums |
022e5c30c079a2d421e164caae244b8e081782c045283d9a061ba18cb22a867f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.13.7
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Feb 28, 2026.
Transparency logSecurity for deepx-dock 0.9.11
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
Full provenance
All 1 file in this release has been uploaded with provenance metadata.
Attestation for deepx_dock-0.9.11-py3-none-any.whl
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Feb 28, 2026.
Transparency logA note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.