diff --git a/.github/common-header.txt b/.github/common-header.txt index cd8a28b..b330aed 100644 --- a/.github/common-header.txt +++ b/.github/common-header.txt @@ -1,2 +1,2 @@ -Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +Copyright (c) 2026 ByteDance Ltd. and/or its affiliates SPDX-License-Identifier: Apache-2.0 diff --git a/docs/_static/export.py b/docs/_static/export.py index 0edfbc4..7eac089 100644 --- a/docs/_static/export.py +++ b/docs/_static/export.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Export design.svg pages to various formats using Inkscape.""" diff --git a/docs/conf.py b/docs/conf.py index 9ff7195..8007512 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 # Configuration file for the Sphinx documentation builder. diff --git a/docs/config_reference.py b/docs/config_reference.py index e9eeed8..11131c5 100644 --- a/docs/config_reference.py +++ b/docs/config_reference.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Sphinx directives for rendering contextual config defaults in docs. diff --git a/docs/jaqmc_sphinx_utils.py b/docs/jaqmc_sphinx_utils.py index 0543f6b..4dc8146 100644 --- a/docs/jaqmc_sphinx_utils.py +++ b/docs/jaqmc_sphinx_utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Picklable typehints formatter for sphinx-autodoc-typehints.""" diff --git a/docs/protocol_autodoc.py b/docs/protocol_autodoc.py index cb4c966..6c127ca 100644 --- a/docs/protocol_autodoc.py +++ b/docs/protocol_autodoc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Minimal Protocol-aware autodoc augmentation for Sphinx. diff --git a/docs/typehints_formatter.py b/docs/typehints_formatter.py index 92b74fa..8be7299 100644 --- a/docs/typehints_formatter.py +++ b/docs/typehints_formatter.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Picklable typehints formatter for sphinx-autodoc-typehints.""" diff --git a/legacy_examples/dmc/deeperwin/deeperwin_model.py b/legacy_examples/dmc/deeperwin/deeperwin_model.py index a5f1d80..9617c81 100644 --- a/legacy_examples/dmc/deeperwin/deeperwin_model.py +++ b/legacy_examples/dmc/deeperwin/deeperwin_model.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/dmc/deeperwin/run.py b/legacy_examples/dmc/deeperwin/run.py index 8c04ec9..35fac8a 100644 --- a/legacy_examples/dmc/deeperwin/run.py +++ b/legacy_examples/dmc/deeperwin/run.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/dmc/dmc_config.py b/legacy_examples/dmc/dmc_config.py index ba35599..0ac8366 100644 --- a/legacy_examples/dmc/dmc_config.py +++ b/legacy_examples/dmc/dmc_config.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/dmc/ferminet/run.py b/legacy_examples/dmc/ferminet/run.py index 6d0c9d6..b73c925 100644 --- a/legacy_examples/dmc/ferminet/run.py +++ b/legacy_examples/dmc/ferminet/run.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/dmc/lapnet/run.py b/legacy_examples/dmc/lapnet/run.py index 211332d..ad53df2 100644 --- a/legacy_examples/dmc/lapnet/run.py +++ b/legacy_examples/dmc/lapnet/run.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/loss/lapnet/atom_spin_state.py b/legacy_examples/loss/lapnet/atom_spin_state.py index 933ddd1..3286732 100644 --- a/legacy_examples/loss/lapnet/atom_spin_state.py +++ b/legacy_examples/loss/lapnet/atom_spin_state.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from lapnet import base_config diff --git a/legacy_examples/loss/lapnet/run.py b/legacy_examples/loss/lapnet/run.py index c871892..c144ad3 100644 --- a/legacy_examples/loss/lapnet/run.py +++ b/legacy_examples/loss/lapnet/run.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/loss/loss_config.py b/legacy_examples/loss/loss_config.py index 03ba8fd..f4213e4 100644 --- a/legacy_examples/loss/loss_config.py +++ b/legacy_examples/loss/loss_config.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/legacy_examples/pp/lapnet/Makefile b/legacy_examples/pp/lapnet/Makefile index eb305d6..48e52b1 100644 --- a/legacy_examples/pp/lapnet/Makefile +++ b/legacy_examples/pp/lapnet/Makefile @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 # =============================== diff --git a/legacy_examples/pp/lapnet/configs/ecp/X.py b/legacy_examples/pp/lapnet/configs/ecp/X.py index 5bd7153..bd94fc6 100644 --- a/legacy_examples/pp/lapnet/configs/ecp/X.py +++ b/legacy_examples/pp/lapnet/configs/ecp/X.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/ecp/XO.py b/legacy_examples/pp/lapnet/configs/ecp/XO.py index f6e77c9..0b26683 100644 --- a/legacy_examples/pp/lapnet/configs/ecp/XO.py +++ b/legacy_examples/pp/lapnet/configs/ecp/XO.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/hph/Sc.py b/legacy_examples/pp/lapnet/configs/hph/Sc.py index 3b05a81..a7fc223 100644 --- a/legacy_examples/pp/lapnet/configs/hph/Sc.py +++ b/legacy_examples/pp/lapnet/configs/hph/Sc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/hph/ScO.py b/legacy_examples/pp/lapnet/configs/hph/ScO.py index 81851fd..f00c43a 100644 --- a/legacy_examples/pp/lapnet/configs/hph/ScO.py +++ b/legacy_examples/pp/lapnet/configs/hph/ScO.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/hph/ScS.py b/legacy_examples/pp/lapnet/configs/hph/ScS.py index ccaa5cd..676e1ce 100644 --- a/legacy_examples/pp/lapnet/configs/hph/ScS.py +++ b/legacy_examples/pp/lapnet/configs/hph/ScS.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/ph/X.py b/legacy_examples/pp/lapnet/configs/ph/X.py index 6778e5e..85c349c 100644 --- a/legacy_examples/pp/lapnet/configs/ph/X.py +++ b/legacy_examples/pp/lapnet/configs/ph/X.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/configs/ph/XO.py b/legacy_examples/pp/lapnet/configs/ph/XO.py index b83bc7c..cabe3b9 100644 --- a/legacy_examples/pp/lapnet/configs/ph/XO.py +++ b/legacy_examples/pp/lapnet/configs/ph/XO.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pyscf import gto diff --git a/legacy_examples/pp/lapnet/run.py b/legacy_examples/pp/lapnet/run.py index e058f32..ddb18ee 100644 --- a/legacy_examples/pp/lapnet/run.py +++ b/legacy_examples/pp/lapnet/run.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc/__init__.py b/src/jaqmc/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/__init__.py +++ b/src/jaqmc/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/app/__init__.py b/src/jaqmc/app/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/app/__init__.py +++ b/src/jaqmc/app/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/app/cli.py b/src/jaqmc/app/cli.py index a62f0fa..36005aa 100644 --- a/src/jaqmc/app/cli.py +++ b/src/jaqmc/app/cli.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """CLI entrypoints for JaQMC applications. diff --git a/src/jaqmc/app/hall/__init__.py b/src/jaqmc/app/hall/__init__.py index afa25a7..4cd092b 100644 --- a/src/jaqmc/app/hall/__init__.py +++ b/src/jaqmc/app/hall/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .workflow import HallEvalWorkflow, HallTrainWorkflow diff --git a/src/jaqmc/app/hall/config.py b/src/jaqmc/app/hall/config.py index ad92787..251c40d 100644 --- a/src/jaqmc/app/hall/config.py +++ b/src/jaqmc/app/hall/config.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Configuration for the quantum Hall effect on a Haldane sphere.""" diff --git a/src/jaqmc/app/hall/data.py b/src/jaqmc/app/hall/data.py index 820fb4e..ea9851f 100644 --- a/src/jaqmc/app/hall/data.py +++ b/src/jaqmc/app/hall/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Data structures for quantum Hall effect simulations.""" diff --git a/src/jaqmc/app/hall/estimator/__init__.py b/src/jaqmc/app/hall/estimator/__init__.py index 4956fef..94b7bfe 100644 --- a/src/jaqmc/app/hall/estimator/__init__.py +++ b/src/jaqmc/app/hall/estimator/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .one_rdm import OneRDM diff --git a/src/jaqmc/app/hall/estimator/one_rdm.py b/src/jaqmc/app/hall/estimator/one_rdm.py index ce819b4..2985f86 100644 --- a/src/jaqmc/app/hall/estimator/one_rdm.py +++ b/src/jaqmc/app/hall/estimator/one_rdm.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""One-body reduced density matrix (1-RDM) estimator for the Haldane sphere. diff --git a/src/jaqmc/app/hall/estimator/pair_correlation.py b/src/jaqmc/app/hall/estimator/pair_correlation.py index 4aeb56c..e2ca429 100644 --- a/src/jaqmc/app/hall/estimator/pair_correlation.py +++ b/src/jaqmc/app/hall/estimator/pair_correlation.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Pair correlation function estimator for the Haldane sphere. diff --git a/src/jaqmc/app/hall/estimator/penalized_loss.py b/src/jaqmc/app/hall/estimator/penalized_loss.py index cf18dfd..0040b19 100644 --- a/src/jaqmc/app/hall/estimator/penalized_loss.py +++ b/src/jaqmc/app/hall/estimator/penalized_loss.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Angular momentum penalty estimator for state selection. diff --git a/src/jaqmc/app/hall/hamiltonian.py b/src/jaqmc/app/hall/hamiltonian.py index 848f637..52e24b9 100644 --- a/src/jaqmc/app/hall/hamiltonian.py +++ b/src/jaqmc/app/hall/hamiltonian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Potential energy estimator for the Haldane sphere.""" diff --git a/src/jaqmc/app/hall/wavefunction/__init__.py b/src/jaqmc/app/hall/wavefunction/__init__.py index d0ba4c1..54be40c 100644 --- a/src/jaqmc/app/hall/wavefunction/__init__.py +++ b/src/jaqmc/app/hall/wavefunction/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .mhpo import MHPO diff --git a/src/jaqmc/app/hall/wavefunction/free.py b/src/jaqmc/app/hall/wavefunction/free.py index 22cea7e..8e026c9 100644 --- a/src/jaqmc/app/hall/wavefunction/free.py +++ b/src/jaqmc/app/hall/wavefunction/free.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Free-electron monopole harmonics wavefunction (for benchmarking).""" diff --git a/src/jaqmc/app/hall/wavefunction/jastrow.py b/src/jaqmc/app/hall/wavefunction/jastrow.py index 1dfe377..51906c7 100644 --- a/src/jaqmc/app/hall/wavefunction/jastrow.py +++ b/src/jaqmc/app/hall/wavefunction/jastrow.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Jastrow factors for wavefunctions on the Haldane sphere.""" diff --git a/src/jaqmc/app/hall/wavefunction/laughlin.py b/src/jaqmc/app/hall/wavefunction/laughlin.py index 7c06d2e..e4ae302 100644 --- a/src/jaqmc/app/hall/wavefunction/laughlin.py +++ b/src/jaqmc/app/hall/wavefunction/laughlin.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Exact Laughlin wavefunction on the Haldane sphere (for benchmarking).""" diff --git a/src/jaqmc/app/hall/wavefunction/mhpo.py b/src/jaqmc/app/hall/wavefunction/mhpo.py index 78f410e..a786275 100644 --- a/src/jaqmc/app/hall/wavefunction/mhpo.py +++ b/src/jaqmc/app/hall/wavefunction/mhpo.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Monopole harmonics product orbital (MHPO) wavefunction on the Haldane sphere. diff --git a/src/jaqmc/app/hall/workflow.py b/src/jaqmc/app/hall/workflow.py index f06bb76..c499645 100644 --- a/src/jaqmc/app/hall/workflow.py +++ b/src/jaqmc/app/hall/workflow.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Workflows for the fractional quantum Hall effect on a Haldane sphere.""" diff --git a/src/jaqmc/app/hydrogen_atom.py b/src/jaqmc/app/hydrogen_atom.py index 33e7f42..a4f9f7b 100644 --- a/src/jaqmc/app/hydrogen_atom.py +++ b/src/jaqmc/app/hydrogen_atom.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc/app/molecule/__init__.py b/src/jaqmc/app/molecule/__init__.py index d034a3d..fd39162 100644 --- a/src/jaqmc/app/molecule/__init__.py +++ b/src/jaqmc/app/molecule/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .workflow import MoleculeEvalWorkflow, MoleculeTrainWorkflow diff --git a/src/jaqmc/app/molecule/config/__init__.py b/src/jaqmc/app/molecule/config/__init__.py index 2680d3b..caa2906 100644 --- a/src/jaqmc/app/molecule/config/__init__.py +++ b/src/jaqmc/app/molecule/config/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import MoleculeConfig diff --git a/src/jaqmc/app/molecule/config/atom.py b/src/jaqmc/app/molecule/config/atom.py index da15389..bbae411 100644 --- a/src/jaqmc/app/molecule/config/atom.py +++ b/src/jaqmc/app/molecule/config/atom.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jaqmc.utils.atomic import make_atom diff --git a/src/jaqmc/app/molecule/config/base.py b/src/jaqmc/app/molecule/config/base.py index e178328..31cbfff 100644 --- a/src/jaqmc/app/molecule/config/base.py +++ b/src/jaqmc/app/molecule/config/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import field diff --git a/src/jaqmc/app/molecule/config/diatomic.py b/src/jaqmc/app/molecule/config/diatomic.py index 058381e..2782b28 100644 --- a/src/jaqmc/app/molecule/config/diatomic.py +++ b/src/jaqmc/app/molecule/config/diatomic.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import re diff --git a/src/jaqmc/app/molecule/config/simple_molecule.py b/src/jaqmc/app/molecule/config/simple_molecule.py index 8cd7e87..b8e77b8 100644 --- a/src/jaqmc/app/molecule/config/simple_molecule.py +++ b/src/jaqmc/app/molecule/config/simple_molecule.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jaqmc.utils.atomic import Atom diff --git a/src/jaqmc/app/molecule/data.py b/src/jaqmc/app/molecule/data.py index 5ce8f7a..4601e8c 100644 --- a/src/jaqmc/app/molecule/data.py +++ b/src/jaqmc/app/molecule/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc/app/molecule/hamiltonian.py b/src/jaqmc/app/molecule/hamiltonian.py index 6aae7ab..23f28b5 100644 --- a/src/jaqmc/app/molecule/hamiltonian.py +++ b/src/jaqmc/app/molecule/hamiltonian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jax import numpy as jnp diff --git a/src/jaqmc/app/molecule/wavefunction/__init__.py b/src/jaqmc/app/molecule/wavefunction/__init__.py index 3e10160..ec025b7 100644 --- a/src/jaqmc/app/molecule/wavefunction/__init__.py +++ b/src/jaqmc/app/molecule/wavefunction/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import MoleculeWavefunction diff --git a/src/jaqmc/app/molecule/wavefunction/base.py b/src/jaqmc/app/molecule/wavefunction/base.py index 1952177..39e23ef 100644 --- a/src/jaqmc/app/molecule/wavefunction/base.py +++ b/src/jaqmc/app/molecule/wavefunction/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Protocol definition for molecular wavefunctions.""" diff --git a/src/jaqmc/app/molecule/wavefunction/ferminet.py b/src/jaqmc/app/molecule/wavefunction/ferminet.py index f27d2ca..bdfeb99 100644 --- a/src/jaqmc/app/molecule/wavefunction/ferminet.py +++ b/src/jaqmc/app/molecule/wavefunction/ferminet.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import field diff --git a/src/jaqmc/app/molecule/wavefunction/psiformer.py b/src/jaqmc/app/molecule/wavefunction/psiformer.py index 29faa5f..5f91c38 100644 --- a/src/jaqmc/app/molecule/wavefunction/psiformer.py +++ b/src/jaqmc/app/molecule/wavefunction/psiformer.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Psiformer wavefunction implementation.""" diff --git a/src/jaqmc/app/molecule/workflow.py b/src/jaqmc/app/molecule/workflow.py index a5c36a5..a3810c7 100644 --- a/src/jaqmc/app/molecule/workflow.py +++ b/src/jaqmc/app/molecule/workflow.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/app/solid/__init__.py b/src/jaqmc/app/solid/__init__.py index 8c550c5..791a0d0 100644 --- a/src/jaqmc/app/solid/__init__.py +++ b/src/jaqmc/app/solid/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .workflow import SolidEvalWorkflow, SolidTrainWorkflow diff --git a/src/jaqmc/app/solid/config/__init__.py b/src/jaqmc/app/solid/config/__init__.py index 2190f80..a603bfa 100644 --- a/src/jaqmc/app/solid/config/__init__.py +++ b/src/jaqmc/app/solid/config/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import SolidConfig diff --git a/src/jaqmc/app/solid/config/base.py b/src/jaqmc/app/solid/config/base.py index b7f90a7..a0e0355 100644 --- a/src/jaqmc/app/solid/config/base.py +++ b/src/jaqmc/app/solid/config/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import field diff --git a/src/jaqmc/app/solid/config/rock_salt.py b/src/jaqmc/app/solid/config/rock_salt.py index 193382a..5e46c68 100644 --- a/src/jaqmc/app/solid/config/rock_salt.py +++ b/src/jaqmc/app/solid/config/rock_salt.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Rock salt (NaCl) crystal structure geometry builder. diff --git a/src/jaqmc/app/solid/config/two_atom_chain.py b/src/jaqmc/app/solid/config/two_atom_chain.py index 4890edb..9112eb7 100644 --- a/src/jaqmc/app/solid/config/two_atom_chain.py +++ b/src/jaqmc/app/solid/config/two_atom_chain.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """1D two-atom chain geometry builder. diff --git a/src/jaqmc/app/solid/data.py b/src/jaqmc/app/solid/data.py index 81cfc31..c9c17a3 100644 --- a/src/jaqmc/app/solid/data.py +++ b/src/jaqmc/app/solid/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc/app/solid/hamiltonian.py b/src/jaqmc/app/solid/hamiltonian.py index 2a0712a..0ce16cd 100644 --- a/src/jaqmc/app/solid/hamiltonian.py +++ b/src/jaqmc/app/solid/hamiltonian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping diff --git a/src/jaqmc/app/solid/wavefunction.py b/src/jaqmc/app/solid/wavefunction.py index 03d6a07..ff1777b 100644 --- a/src/jaqmc/app/solid/wavefunction.py +++ b/src/jaqmc/app/solid/wavefunction.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import field diff --git a/src/jaqmc/app/solid/workflow.py b/src/jaqmc/app/solid/workflow.py index dd84b64..b18f60a 100644 --- a/src/jaqmc/app/solid/workflow.py +++ b/src/jaqmc/app/solid/workflow.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/array_types.py b/src/jaqmc/array_types.py index e7aa9fa..5cbf580 100644 --- a/src/jaqmc/array_types.py +++ b/src/jaqmc/array_types.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping, Sequence diff --git a/src/jaqmc/data.py b/src/jaqmc/data.py index 3384a3a..c98f56a 100644 --- a/src/jaqmc/data.py +++ b/src/jaqmc/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import dataclasses diff --git a/src/jaqmc/estimator/__init__.py b/src/jaqmc/estimator/__init__.py index 84089dd..7eef841 100644 --- a/src/jaqmc/estimator/__init__.py +++ b/src/jaqmc/estimator/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import ( diff --git a/src/jaqmc/estimator/base.py b/src/jaqmc/estimator/base.py index 15e5513..3399f9b 100644 --- a/src/jaqmc/estimator/base.py +++ b/src/jaqmc/estimator/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from abc import ABC, abstractmethod diff --git a/src/jaqmc/estimator/density/__init__.py b/src/jaqmc/estimator/density/__init__.py index 786cd18..2986c3b 100644 --- a/src/jaqmc/estimator/density/__init__.py +++ b/src/jaqmc/estimator/density/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Electron density histogram estimators for different coordinate systems.""" diff --git a/src/jaqmc/estimator/density/cartesian.py b/src/jaqmc/estimator/density/cartesian.py index f3d7958..3616a01 100644 --- a/src/jaqmc/estimator/density/cartesian.py +++ b/src/jaqmc/estimator/density/cartesian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Cartesian density estimator for molecules and open-boundary systems.""" diff --git a/src/jaqmc/estimator/density/fractional.py b/src/jaqmc/estimator/density/fractional.py index 92a7b61..e7bbc60 100644 --- a/src/jaqmc/estimator/density/fractional.py +++ b/src/jaqmc/estimator/density/fractional.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Fractional-coordinate density estimator for periodic systems.""" diff --git a/src/jaqmc/estimator/density/spherical.py b/src/jaqmc/estimator/density/spherical.py index 7745771..a1a8e8a 100644 --- a/src/jaqmc/estimator/density/spherical.py +++ b/src/jaqmc/estimator/density/spherical.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Spherical density estimator for the Haldane sphere.""" diff --git a/src/jaqmc/estimator/ecp/__init__.py b/src/jaqmc/estimator/ecp/__init__.py index 1ecc4ca..d9bd714 100644 --- a/src/jaqmc/estimator/ecp/__init__.py +++ b/src/jaqmc/estimator/ecp/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Effective Core Potential (ECP) estimator for molecular QMC. diff --git a/src/jaqmc/estimator/ecp/estimator.py b/src/jaqmc/estimator/ecp/estimator.py index 7da3d6b..cb616e1 100644 --- a/src/jaqmc/estimator/ecp/estimator.py +++ b/src/jaqmc/estimator/ecp/estimator.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""ECP energy estimator for molecular QMC calculations. diff --git a/src/jaqmc/estimator/ecp/nonlocal_integral.py b/src/jaqmc/estimator/ecp/nonlocal_integral.py index 380f1f4..206e690 100644 --- a/src/jaqmc/estimator/ecp/nonlocal_integral.py +++ b/src/jaqmc/estimator/ecp/nonlocal_integral.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Nonlocal ECP integration using spherical quadrature. diff --git a/src/jaqmc/estimator/ecp/quadrature.py b/src/jaqmc/estimator/ecp/quadrature.py index 7d57cec..a295ecd 100644 --- a/src/jaqmc/estimator/ecp/quadrature.py +++ b/src/jaqmc/estimator/ecp/quadrature.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Spherical quadrature rules for ECP nonlocal integration. diff --git a/src/jaqmc/estimator/ewald.py b/src/jaqmc/estimator/ewald.py index b2b44e1..c987167 100644 --- a/src/jaqmc/estimator/ewald.py +++ b/src/jaqmc/estimator/ewald.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/estimator/histogram.py b/src/jaqmc/estimator/histogram.py index 1ba953c..6be492f 100644 --- a/src/jaqmc/estimator/histogram.py +++ b/src/jaqmc/estimator/histogram.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Base class for histogram-accumulating estimators with Kahan summation.""" diff --git a/src/jaqmc/estimator/kinetic/__init__.py b/src/jaqmc/estimator/kinetic/__init__.py index fd3b6b1..8191850 100644 --- a/src/jaqmc/estimator/kinetic/__init__.py +++ b/src/jaqmc/estimator/kinetic/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Kinetic energy estimators for Euclidean and spherical geometries.""" diff --git a/src/jaqmc/estimator/kinetic/_common.py b/src/jaqmc/estimator/kinetic/_common.py index 0f13f60..9f31a02 100644 --- a/src/jaqmc/estimator/kinetic/_common.py +++ b/src/jaqmc/estimator/kinetic/_common.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Shared utilities for kinetic energy estimators.""" diff --git a/src/jaqmc/estimator/kinetic/euclidean.py b/src/jaqmc/estimator/kinetic/euclidean.py index 888687b..27a7080 100644 --- a/src/jaqmc/estimator/kinetic/euclidean.py +++ b/src/jaqmc/estimator/kinetic/euclidean.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Kinetic energy estimator in Euclidean geometry.""" diff --git a/src/jaqmc/estimator/kinetic/spherical.py b/src/jaqmc/estimator/kinetic/spherical.py index ad0541d..04447ce 100644 --- a/src/jaqmc/estimator/kinetic/spherical.py +++ b/src/jaqmc/estimator/kinetic/spherical.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Kinetic energy estimator on a sphere with magnetic monopole. diff --git a/src/jaqmc/estimator/loss_grad.py b/src/jaqmc/estimator/loss_grad.py index 5d0f956..d128ac1 100644 --- a/src/jaqmc/estimator/loss_grad.py +++ b/src/jaqmc/estimator/loss_grad.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping diff --git a/src/jaqmc/estimator/ph/__init__.py b/src/jaqmc/estimator/ph/__init__.py index ba05dd0..0ea3df6 100644 --- a/src/jaqmc/estimator/ph/__init__.py +++ b/src/jaqmc/estimator/ph/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Pseudo-Hamiltonian estimator.""" diff --git a/src/jaqmc/estimator/ph/_forward_laplacian.py b/src/jaqmc/estimator/ph/_forward_laplacian.py index 4d260fa..3d5ca37 100644 --- a/src/jaqmc/estimator/ph/_forward_laplacian.py +++ b/src/jaqmc/estimator/ph/_forward_laplacian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Forward-Laplacian backend for the PH derivative term — production path. diff --git a/src/jaqmc/estimator/ph/_standard.py b/src/jaqmc/estimator/ph/_standard.py index e1e600d..d56d8c1 100644 --- a/src/jaqmc/estimator/ph/_standard.py +++ b/src/jaqmc/estimator/ph/_standard.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Standard PH derivative backend — reverse-mode reference implementation. diff --git a/src/jaqmc/estimator/ph/data.py b/src/jaqmc/estimator/ph/data.py index 079dcf7..4f422e6 100644 --- a/src/jaqmc/estimator/ph/data.py +++ b/src/jaqmc/estimator/ph/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """PH reference data loader. diff --git a/src/jaqmc/estimator/ph/estimator.py b/src/jaqmc/estimator/ph/estimator.py index a3e122a..d253c43 100644 --- a/src/jaqmc/estimator/ph/estimator.py +++ b/src/jaqmc/estimator/ph/estimator.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Pseudo-Hamiltonian energy estimator. diff --git a/src/jaqmc/estimator/spin.py b/src/jaqmc/estimator/spin.py index 89aeca8..c23227e 100644 --- a/src/jaqmc/estimator/spin.py +++ b/src/jaqmc/estimator/spin.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Spin-squared (:math:`S^2`) estimator. diff --git a/src/jaqmc/estimator/total_energy.py b/src/jaqmc/estimator/total_energy.py index 257cada..64778ef 100644 --- a/src/jaqmc/estimator/total_energy.py +++ b/src/jaqmc/estimator/total_energy.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping diff --git a/src/jaqmc/geometry/__init__.py b/src/jaqmc/geometry/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/geometry/__init__.py +++ b/src/jaqmc/geometry/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/geometry/obc.py b/src/jaqmc/geometry/obc.py index 669de12..0afb821 100644 --- a/src/jaqmc/geometry/obc.py +++ b/src/jaqmc/geometry/obc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jax import numpy as jnp diff --git a/src/jaqmc/geometry/pbc.py b/src/jaqmc/geometry/pbc.py index 202f6e1..942f8d6 100644 --- a/src/jaqmc/geometry/pbc.py +++ b/src/jaqmc/geometry/pbc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from enum import StrEnum diff --git a/src/jaqmc/geometry/sphere.py b/src/jaqmc/geometry/sphere.py index a612598..7339ad7 100644 --- a/src/jaqmc/geometry/sphere.py +++ b/src/jaqmc/geometry/sphere.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Spherical geometry utilities for MCMC sampling on the sphere.""" diff --git a/src/jaqmc/optimizer/__init__.py b/src/jaqmc/optimizer/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/optimizer/__init__.py +++ b/src/jaqmc/optimizer/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/optimizer/base.py b/src/jaqmc/optimizer/base.py index 2acafd0..8342ae6 100644 --- a/src/jaqmc/optimizer/base.py +++ b/src/jaqmc/optimizer/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Protocol, runtime_checkable diff --git a/src/jaqmc/optimizer/kfac/__init__.py b/src/jaqmc/optimizer/kfac/__init__.py index 06fc41f..2c262e2 100644 --- a/src/jaqmc/optimizer/kfac/__init__.py +++ b/src/jaqmc/optimizer/kfac/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .kfac import KFACOptimizer as KFACOptimizer diff --git a/src/jaqmc/optimizer/kfac/complex_support.py b/src/jaqmc/optimizer/kfac/complex_support.py index 1ef5a7f..862b4b9 100644 --- a/src/jaqmc/optimizer/kfac/complex_support.py +++ b/src/jaqmc/optimizer/kfac/complex_support.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Make kfac_jax be able to handle stochastic reconfiguration. diff --git a/src/jaqmc/optimizer/kfac/curvature_blocks.py b/src/jaqmc/optimizer/kfac/curvature_blocks.py index 5008d4e..b22631c 100644 --- a/src/jaqmc/optimizer/kfac/curvature_blocks.py +++ b/src/jaqmc/optimizer/kfac/curvature_blocks.py @@ -1,5 +1,5 @@ # Copyright 2020 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/optimizer/kfac/kfac.py b/src/jaqmc/optimizer/kfac/kfac.py index 202ee92..913eab9 100644 --- a/src/jaqmc/optimizer/kfac/kfac.py +++ b/src/jaqmc/optimizer/kfac/kfac.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Any, NamedTuple diff --git a/src/jaqmc/optimizer/kfac/tag_registration.py b/src/jaqmc/optimizer/kfac/tag_registration.py index 65f2e14..99c5169 100644 --- a/src/jaqmc/optimizer/kfac/tag_registration.py +++ b/src/jaqmc/optimizer/kfac/tag_registration.py @@ -1,5 +1,5 @@ # Copyright 2020 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/optimizer/optax.py b/src/jaqmc/optimizer/optax.py index d887da3..a727d03 100644 --- a/src/jaqmc/optimizer/optax.py +++ b/src/jaqmc/optimizer/optax.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import inspect diff --git a/src/jaqmc/optimizer/schedule.py b/src/jaqmc/optimizer/schedule.py index 78f2ec7..0e83823 100644 --- a/src/jaqmc/optimizer/schedule.py +++ b/src/jaqmc/optimizer/schedule.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jax import numpy as jnp diff --git a/src/jaqmc/optimizer/sr/__init__.py b/src/jaqmc/optimizer/sr/__init__.py index baf0572..388e968 100644 --- a/src/jaqmc/optimizer/sr/__init__.py +++ b/src/jaqmc/optimizer/sr/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .sr import SROptimizer as SROptimizer diff --git a/src/jaqmc/optimizer/sr/jasr.py b/src/jaqmc/optimizer/sr/jasr.py index a68ad52..73f39df 100644 --- a/src/jaqmc/optimizer/sr/jasr.py +++ b/src/jaqmc/optimizer/sr/jasr.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import dataclasses diff --git a/src/jaqmc/optimizer/sr/sr.py b/src/jaqmc/optimizer/sr/sr.py index e43efec..a2a6a32 100644 --- a/src/jaqmc/optimizer/sr/sr.py +++ b/src/jaqmc/optimizer/sr/sr.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Any, Literal diff --git a/src/jaqmc/sampler/__init__.py b/src/jaqmc/sampler/__init__.py index 21ab621..2d283c7 100644 --- a/src/jaqmc/sampler/__init__.py +++ b/src/jaqmc/sampler/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Samplers updates (parts of) the data according to a distribution or data pool. diff --git a/src/jaqmc/sampler/base.py b/src/jaqmc/sampler/base.py index b66e949..6c1599d 100644 --- a/src/jaqmc/sampler/base.py +++ b/src/jaqmc/sampler/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping diff --git a/src/jaqmc/sampler/mcmc.py b/src/jaqmc/sampler/mcmc.py index 1b77c4c..359e830 100644 --- a/src/jaqmc/sampler/mcmc.py +++ b/src/jaqmc/sampler/mcmc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Any, NamedTuple, Protocol diff --git a/src/jaqmc/utils/__init__.py b/src/jaqmc/utils/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/utils/__init__.py +++ b/src/jaqmc/utils/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/utils/array.py b/src/jaqmc/utils/array.py index 30f4e1f..4ab639a 100644 --- a/src/jaqmc/utils/array.py +++ b/src/jaqmc/utils/array.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import itertools diff --git a/src/jaqmc/utils/atomic/__init__.py b/src/jaqmc/utils/atomic/__init__.py index 3becbc8..757d324 100644 --- a/src/jaqmc/utils/atomic/__init__.py +++ b/src/jaqmc/utils/atomic/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .atom import Atom diff --git a/src/jaqmc/utils/atomic/atom.py b/src/jaqmc/utils/atomic/atom.py index eaf79fd..f36f5d2 100644 --- a/src/jaqmc/utils/atomic/atom.py +++ b/src/jaqmc/utils/atomic/atom.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jax import numpy as jnp diff --git a/src/jaqmc/utils/atomic/atomic_system.py b/src/jaqmc/utils/atomic/atomic_system.py index e419eba..e6f3120 100644 --- a/src/jaqmc/utils/atomic/atomic_system.py +++ b/src/jaqmc/utils/atomic/atomic_system.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jaqmc.utils.atomic.atom import Atom diff --git a/src/jaqmc/utils/atomic/elements.py b/src/jaqmc/utils/atomic/elements.py index 287014b..caf1166 100644 --- a/src/jaqmc/utils/atomic/elements.py +++ b/src/jaqmc/utils/atomic/elements.py @@ -1,5 +1,5 @@ # Copyright 2019 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/utils/atomic/factory.py b/src/jaqmc/utils/atomic/factory.py index 816518c..b2989de 100644 --- a/src/jaqmc/utils/atomic/factory.py +++ b/src/jaqmc/utils/atomic/factory.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Factory helpers for system-layer atomic objects. diff --git a/src/jaqmc/utils/atomic/gto.py b/src/jaqmc/utils/atomic/gto.py index 26ee03c..ef8f34e 100644 --- a/src/jaqmc/utils/atomic/gto.py +++ b/src/jaqmc/utils/atomic/gto.py @@ -1,5 +1,5 @@ # Copyright 2023 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/utils/atomic/initialization.py b/src/jaqmc/utils/atomic/initialization.py index 0a3ce56..713055e 100644 --- a/src/jaqmc/utils/atomic/initialization.py +++ b/src/jaqmc/utils/atomic/initialization.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc/utils/atomic/pp.py b/src/jaqmc/utils/atomic/pp.py index b17455a..1a6662b 100644 --- a/src/jaqmc/utils/atomic/pp.py +++ b/src/jaqmc/utils/atomic/pp.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Pseudopotential vocabulary and helpers. diff --git a/src/jaqmc/utils/atomic/pretrain.py b/src/jaqmc/utils/atomic/pretrain.py index 4233055..6c68d04 100644 --- a/src/jaqmc/utils/atomic/pretrain.py +++ b/src/jaqmc/utils/atomic/pretrain.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Shared pretrain loss estimator for atomic systems (molecule, solid).""" diff --git a/src/jaqmc/utils/atomic/scf.py b/src/jaqmc/utils/atomic/scf.py index 707b948..c27a5e3 100644 --- a/src/jaqmc/utils/atomic/scf.py +++ b/src/jaqmc/utils/atomic/scf.py @@ -1,5 +1,5 @@ # Copyright 2019 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/utils/checkpoint.py b/src/jaqmc/utils/checkpoint.py index 820a363..c1e0f66 100644 --- a/src/jaqmc/utils/checkpoint.py +++ b/src/jaqmc/utils/checkpoint.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/utils/chunked_vmap.py b/src/jaqmc/utils/chunked_vmap.py index 211f850..48db64a 100644 --- a/src/jaqmc/utils/chunked_vmap.py +++ b/src/jaqmc/utils/chunked_vmap.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc/utils/cli.py b/src/jaqmc/utils/cli.py index 8b429b5..8d8b5d5 100644 --- a/src/jaqmc/utils/cli.py +++ b/src/jaqmc/utils/cli.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Sequence diff --git a/src/jaqmc/utils/clip.py b/src/jaqmc/utils/clip.py index 5632b35..62f7f14 100644 --- a/src/jaqmc/utils/clip.py +++ b/src/jaqmc/utils/clip.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Literal diff --git a/src/jaqmc/utils/config.py b/src/jaqmc/utils/config.py index f0a7be8..1942f81 100644 --- a/src/jaqmc/utils/config.py +++ b/src/jaqmc/utils/config.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import difflib diff --git a/src/jaqmc/utils/func_transform.py b/src/jaqmc/utils/func_transform.py index bfca6e5..7149859 100644 --- a/src/jaqmc/utils/func_transform.py +++ b/src/jaqmc/utils/func_transform.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Callable, Mapping, Sequence diff --git a/src/jaqmc/utils/jax_dataclass.py b/src/jaqmc/utils/jax_dataclass.py index 5bc483e..9f57dc4 100644 --- a/src/jaqmc/utils/jax_dataclass.py +++ b/src/jaqmc/utils/jax_dataclass.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import dataclasses diff --git a/src/jaqmc/utils/module_resolver.py b/src/jaqmc/utils/module_resolver.py index 7801b4e..335c032 100644 --- a/src/jaqmc/utils/module_resolver.py +++ b/src/jaqmc/utils/module_resolver.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import importlib.util diff --git a/src/jaqmc/utils/parallel_jax.py b/src/jaqmc/utils/parallel_jax.py index ba833e9..432ed09 100644 --- a/src/jaqmc/utils/parallel_jax.py +++ b/src/jaqmc/utils/parallel_jax.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/utils/runtime.py b/src/jaqmc/utils/runtime.py index 82e769a..2e5788f 100644 --- a/src/jaqmc/utils/runtime.py +++ b/src/jaqmc/utils/runtime.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Startup-time runtime configuration for JaQMC applications.""" diff --git a/src/jaqmc/utils/signal_handler.py b/src/jaqmc/utils/signal_handler.py index 49e8d17..3487e0c 100644 --- a/src/jaqmc/utils/signal_handler.py +++ b/src/jaqmc/utils/signal_handler.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import signal diff --git a/src/jaqmc/utils/supercell.py b/src/jaqmc/utils/supercell.py index 5f27f2a..6502816 100644 --- a/src/jaqmc/utils/supercell.py +++ b/src/jaqmc/utils/supercell.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Literal diff --git a/src/jaqmc/utils/time_tracker.py b/src/jaqmc/utils/time_tracker.py index 1ceb93c..ba968c0 100644 --- a/src/jaqmc/utils/time_tracker.py +++ b/src/jaqmc/utils/time_tracker.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Time-per-step tracker with configurable warmup-step exclusion.""" diff --git a/src/jaqmc/utils/units.py b/src/jaqmc/utils/units.py index be61f65..b30f610 100644 --- a/src/jaqmc/utils/units.py +++ b/src/jaqmc/utils/units.py @@ -1,5 +1,5 @@ # Copyright 2019 DeepMind Technologies Limited. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc/utils/wiring.py b/src/jaqmc/utils/wiring.py index 022901a..5a97943 100644 --- a/src/jaqmc/utils/wiring.py +++ b/src/jaqmc/utils/wiring.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Generic runtime dependency wiring for dataclass components. diff --git a/src/jaqmc/utils/yaml_format.py b/src/jaqmc/utils/yaml_format.py index b79dfe4..a063570 100644 --- a/src/jaqmc/utils/yaml_format.py +++ b/src/jaqmc/utils/yaml_format.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Utilities for YAML processing and annotation.""" diff --git a/src/jaqmc/wavefunction/__init__.py b/src/jaqmc/wavefunction/__init__.py index 1d8ada8..9d5d2f2 100644 --- a/src/jaqmc/wavefunction/__init__.py +++ b/src/jaqmc/wavefunction/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import ( diff --git a/src/jaqmc/wavefunction/backbone/__init__.py b/src/jaqmc/wavefunction/backbone/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/wavefunction/backbone/__init__.py +++ b/src/jaqmc/wavefunction/backbone/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/wavefunction/backbone/ferminet.py b/src/jaqmc/wavefunction/backbone/ferminet.py index d9b7ec4..b6deb98 100644 --- a/src/jaqmc/wavefunction/backbone/ferminet.py +++ b/src/jaqmc/wavefunction/backbone/ferminet.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from flax import linen as nn diff --git a/src/jaqmc/wavefunction/backbone/psiformer.py b/src/jaqmc/wavefunction/backbone/psiformer.py index 7fad0f0..f11eed2 100644 --- a/src/jaqmc/wavefunction/backbone/psiformer.py +++ b/src/jaqmc/wavefunction/backbone/psiformer.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Sequence diff --git a/src/jaqmc/wavefunction/base.py b/src/jaqmc/wavefunction/base.py index c4930a1..72bf404 100644 --- a/src/jaqmc/wavefunction/base.py +++ b/src/jaqmc/wavefunction/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from abc import ABC, abstractmethod diff --git a/src/jaqmc/wavefunction/input/__init__.py b/src/jaqmc/wavefunction/input/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/wavefunction/input/__init__.py +++ b/src/jaqmc/wavefunction/input/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/wavefunction/input/atomic.py b/src/jaqmc/wavefunction/input/atomic.py index ffd4ddd..d350730 100644 --- a/src/jaqmc/wavefunction/input/atomic.py +++ b/src/jaqmc/wavefunction/input/atomic.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import TypedDict diff --git a/src/jaqmc/wavefunction/jastrow.py b/src/jaqmc/wavefunction/jastrow.py index 108f63e..6df0f80 100644 --- a/src/jaqmc/wavefunction/jastrow.py +++ b/src/jaqmc/wavefunction/jastrow.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from flax import linen as nn diff --git a/src/jaqmc/wavefunction/output/__init__.py b/src/jaqmc/wavefunction/output/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc/wavefunction/output/__init__.py +++ b/src/jaqmc/wavefunction/output/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc/wavefunction/output/envelope.py b/src/jaqmc/wavefunction/output/envelope.py index 52a33dd..8861a8a 100644 --- a/src/jaqmc/wavefunction/output/envelope.py +++ b/src/jaqmc/wavefunction/output/envelope.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/wavefunction/output/logdet.py b/src/jaqmc/wavefunction/output/logdet.py index 1675b0e..67d3243 100644 --- a/src/jaqmc/wavefunction/output/logdet.py +++ b/src/jaqmc/wavefunction/output/logdet.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import TypedDict diff --git a/src/jaqmc/wavefunction/output/orbital.py b/src/jaqmc/wavefunction/output/orbital.py index ce3f9c0..a22a4ff 100644 --- a/src/jaqmc/wavefunction/output/orbital.py +++ b/src/jaqmc/wavefunction/output/orbital.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from flax import linen as nn diff --git a/src/jaqmc/workflow/__init__.py b/src/jaqmc/workflow/__init__.py index 3ff9ab1..474c415 100644 --- a/src/jaqmc/workflow/__init__.py +++ b/src/jaqmc/workflow/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import Workflow, WorkflowConfig, init_batched_data diff --git a/src/jaqmc/workflow/base.py b/src/jaqmc/workflow/base.py index 88aa9ef..17c24fb 100644 --- a/src/jaqmc/workflow/base.py +++ b/src/jaqmc/workflow/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import datetime diff --git a/src/jaqmc/workflow/evaluation.py b/src/jaqmc/workflow/evaluation.py index 9a6e7bd..b07e094 100644 --- a/src/jaqmc/workflow/evaluation.py +++ b/src/jaqmc/workflow/evaluation.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Evaluation workflow that loads trained params from a checkpoint.""" diff --git a/src/jaqmc/workflow/stage/__init__.py b/src/jaqmc/workflow/stage/__init__.py index a3eea9d..824278a 100644 --- a/src/jaqmc/workflow/stage/__init__.py +++ b/src/jaqmc/workflow/stage/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Work stage implementations for JaQMC workflows.""" diff --git a/src/jaqmc/workflow/stage/base.py b/src/jaqmc/workflow/stage/base.py index e1a2a55..425dc54 100644 --- a/src/jaqmc/workflow/stage/base.py +++ b/src/jaqmc/workflow/stage/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Shared types and configuration for work stages.""" diff --git a/src/jaqmc/workflow/stage/evaluation.py b/src/jaqmc/workflow/stage/evaluation.py index dd79790..6227124 100644 --- a/src/jaqmc/workflow/stage/evaluation.py +++ b/src/jaqmc/workflow/stage/evaluation.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Evaluation work stage for sampling and observable estimation. diff --git a/src/jaqmc/workflow/stage/sampling.py b/src/jaqmc/workflow/stage/sampling.py index 352a74b..6cd63d0 100644 --- a/src/jaqmc/workflow/stage/sampling.py +++ b/src/jaqmc/workflow/stage/sampling.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Base builder and stage for sampling-based work stages.""" diff --git a/src/jaqmc/workflow/stage/vmc.py b/src/jaqmc/workflow/stage/vmc.py index ef7516f..502b5ff 100644 --- a/src/jaqmc/workflow/stage/vmc.py +++ b/src/jaqmc/workflow/stage/vmc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Variational Monte Carlo work stage for sampling and training.""" diff --git a/src/jaqmc/workflow/vmc.py b/src/jaqmc/workflow/vmc.py index ab82d0d..aeb88db 100644 --- a/src/jaqmc/workflow/vmc.py +++ b/src/jaqmc/workflow/vmc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import time diff --git a/src/jaqmc/writer/__init__.py b/src/jaqmc/writer/__init__.py index 34cd0f9..57fcf0e 100644 --- a/src/jaqmc/writer/__init__.py +++ b/src/jaqmc/writer/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .base import Writer, Writers diff --git a/src/jaqmc/writer/base.py b/src/jaqmc/writer/base.py index 349ec0c..c41264b 100644 --- a/src/jaqmc/writer/base.py +++ b/src/jaqmc/writer/base.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from abc import ABC, abstractmethod diff --git a/src/jaqmc/writer/console.py b/src/jaqmc/writer/console.py index 51191c7..bf0e798 100644 --- a/src/jaqmc/writer/console.py +++ b/src/jaqmc/writer/console.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import logging diff --git a/src/jaqmc/writer/csv.py b/src/jaqmc/writer/csv.py index 4d4fb68..b8fb8ca 100644 --- a/src/jaqmc/writer/csv.py +++ b/src/jaqmc/writer/csv.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import csv diff --git a/src/jaqmc/writer/hdf5.py b/src/jaqmc/writer/hdf5.py index 6300c36..b45fa9a 100644 --- a/src/jaqmc/writer/hdf5.py +++ b/src/jaqmc/writer/hdf5.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from collections.abc import Mapping diff --git a/src/jaqmc/writer/wandb.py b/src/jaqmc/writer/wandb.py index b402f44..fcbf3cd 100644 --- a/src/jaqmc/writer/wandb.py +++ b/src/jaqmc/writer/wandb.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import importlib diff --git a/src/jaqmc_legacy/__init__.py b/src/jaqmc_legacy/__init__.py index c49c678..e63d822 100644 --- a/src/jaqmc_legacy/__init__.py +++ b/src/jaqmc_legacy/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc_legacy/dmc/__init__.py b/src/jaqmc_legacy/dmc/__init__.py index 0070a82..38b8036 100644 --- a/src/jaqmc_legacy/dmc/__init__.py +++ b/src/jaqmc_legacy/dmc/__init__.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from .dmc import run diff --git a/src/jaqmc_legacy/dmc/branch.py b/src/jaqmc_legacy/dmc/branch.py index 4b2cb36..0e206dc 100644 --- a/src/jaqmc_legacy/dmc/branch.py +++ b/src/jaqmc_legacy/dmc/branch.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/branch_fix_size.py b/src/jaqmc_legacy/dmc/branch_fix_size.py index 7ea9a64..3624742 100644 --- a/src/jaqmc_legacy/dmc/branch_fix_size.py +++ b/src/jaqmc_legacy/dmc/branch_fix_size.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/ckpt_handler.py b/src/jaqmc_legacy/dmc/ckpt_handler.py index 19d2252..e4363bc 100644 --- a/src/jaqmc_legacy/dmc/ckpt_handler.py +++ b/src/jaqmc_legacy/dmc/ckpt_handler.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/ckpt_metric_manager.py b/src/jaqmc_legacy/dmc/ckpt_metric_manager.py index 599c283..1e25460 100644 --- a/src/jaqmc_legacy/dmc/ckpt_metric_manager.py +++ b/src/jaqmc_legacy/dmc/ckpt_metric_manager.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/data_path.py b/src/jaqmc_legacy/dmc/data_path.py index ca164f5..615b4d0 100644 --- a/src/jaqmc_legacy/dmc/data_path.py +++ b/src/jaqmc_legacy/dmc/data_path.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/dmc.py b/src/jaqmc_legacy/dmc/dmc.py index 725aafc..41da97f 100644 --- a/src/jaqmc_legacy/dmc/dmc.py +++ b/src/jaqmc_legacy/dmc/dmc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/effective_time_step_calculator.py b/src/jaqmc_legacy/dmc/effective_time_step_calculator.py index 083c9a7..8de59ea 100644 --- a/src/jaqmc_legacy/dmc/effective_time_step_calculator.py +++ b/src/jaqmc_legacy/dmc/effective_time_step_calculator.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/energy_estimator.py b/src/jaqmc_legacy/dmc/energy_estimator.py index 177fb85..95ecda0 100644 --- a/src/jaqmc_legacy/dmc/energy_estimator.py +++ b/src/jaqmc_legacy/dmc/energy_estimator.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/hamiltonian.py b/src/jaqmc_legacy/dmc/hamiltonian.py index 4baa2b6..b63fdbc 100644 --- a/src/jaqmc_legacy/dmc/hamiltonian.py +++ b/src/jaqmc_legacy/dmc/hamiltonian.py @@ -1,5 +1,5 @@ # Copyright 2020 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2022. diff --git a/src/jaqmc_legacy/dmc/metric_manager.py b/src/jaqmc_legacy/dmc/metric_manager.py index 1b2d87e..baac681 100644 --- a/src/jaqmc_legacy/dmc/metric_manager.py +++ b/src/jaqmc_legacy/dmc/metric_manager.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/position_update.py b/src/jaqmc_legacy/dmc/position_update.py index 82b7deb..8f4a348 100644 --- a/src/jaqmc_legacy/dmc/position_update.py +++ b/src/jaqmc_legacy/dmc/position_update.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/state.py b/src/jaqmc_legacy/dmc/state.py index d2434e0..01f7cf8 100644 --- a/src/jaqmc_legacy/dmc/state.py +++ b/src/jaqmc_legacy/dmc/state.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/storage_handler.py b/src/jaqmc_legacy/dmc/storage_handler.py index a053160..d662335 100644 --- a/src/jaqmc_legacy/dmc/storage_handler.py +++ b/src/jaqmc_legacy/dmc/storage_handler.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/dmc/utils.py b/src/jaqmc_legacy/dmc/utils.py index 1f181b5..89fb36c 100644 --- a/src/jaqmc_legacy/dmc/utils.py +++ b/src/jaqmc_legacy/dmc/utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2022-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 ''' diff --git a/src/jaqmc_legacy/loss/__init__.py b/src/jaqmc_legacy/loss/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc_legacy/loss/__init__.py +++ b/src/jaqmc_legacy/loss/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc_legacy/loss/factory.py b/src/jaqmc_legacy/loss/factory.py index e9d9e7d..a4ecf69 100644 --- a/src/jaqmc_legacy/loss/factory.py +++ b/src/jaqmc_legacy/loss/factory.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import inspect diff --git a/src/jaqmc_legacy/loss/overlap_penalty.py b/src/jaqmc_legacy/loss/overlap_penalty.py index 2e669de..dc35d65 100644 --- a/src/jaqmc_legacy/loss/overlap_penalty.py +++ b/src/jaqmc_legacy/loss/overlap_penalty.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import List, Tuple diff --git a/src/jaqmc_legacy/loss/spin_penalty.py b/src/jaqmc_legacy/loss/spin_penalty.py index 459fb04..3d818cf 100644 --- a/src/jaqmc_legacy/loss/spin_penalty.py +++ b/src/jaqmc_legacy/loss/spin_penalty.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Tuple diff --git a/src/jaqmc_legacy/loss/utils.py b/src/jaqmc_legacy/loss/utils.py index 8b1101f..6d44d56 100644 --- a/src/jaqmc_legacy/loss/utils.py +++ b/src/jaqmc_legacy/loss/utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import functools diff --git a/src/jaqmc_legacy/loss/vmc.py b/src/jaqmc_legacy/loss/vmc.py index 2e97207..9928bf6 100644 --- a/src/jaqmc_legacy/loss/vmc.py +++ b/src/jaqmc_legacy/loss/vmc.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2024-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Tuple diff --git a/src/jaqmc_legacy/pp/__init__.py b/src/jaqmc_legacy/pp/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc_legacy/pp/__init__.py +++ b/src/jaqmc_legacy/pp/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc_legacy/pp/ecp/data.py b/src/jaqmc_legacy/pp/ecp/data.py index 05f39fd..6b8244b 100644 --- a/src/jaqmc_legacy/pp/ecp/data.py +++ b/src/jaqmc_legacy/pp/ecp/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pathlib import Path diff --git a/src/jaqmc_legacy/pp/ecp_potential.py b/src/jaqmc_legacy/pp/ecp_potential.py index afccf90..1cf18cb 100644 --- a/src/jaqmc_legacy/pp/ecp_potential.py +++ b/src/jaqmc_legacy/pp/ecp_potential.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc_legacy/pp/hamiltonian.py b/src/jaqmc_legacy/pp/hamiltonian.py index 528f28b..6b1e6b8 100644 --- a/src/jaqmc_legacy/pp/hamiltonian.py +++ b/src/jaqmc_legacy/pp/hamiltonian.py @@ -1,5 +1,5 @@ # Copyright 2020 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/src/jaqmc_legacy/pp/ph/__init__.py b/src/jaqmc_legacy/pp/ph/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc_legacy/pp/ph/__init__.py +++ b/src/jaqmc_legacy/pp/ph/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc_legacy/pp/ph/data.py b/src/jaqmc_legacy/pp/ph/data.py index 52d76d8..bfb292c 100644 --- a/src/jaqmc_legacy/pp/ph/data.py +++ b/src/jaqmc_legacy/pp/ph/data.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/src/jaqmc_legacy/pp/ph/hamiltonian.py b/src/jaqmc_legacy/pp/ph/hamiltonian.py index 0238174..e9be751 100644 --- a/src/jaqmc_legacy/pp/ph/hamiltonian.py +++ b/src/jaqmc_legacy/pp/ph/hamiltonian.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/src/jaqmc_legacy/pp/pp_config.py b/src/jaqmc_legacy/pp/pp_config.py index 23c9b84..e40be2a 100644 --- a/src/jaqmc_legacy/pp/pp_config.py +++ b/src/jaqmc_legacy/pp/pp_config.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import ml_collections diff --git a/src/jaqmc_legacy/pp/quadrature.py b/src/jaqmc_legacy/pp/quadrature.py index 868535a..5588bb0 100644 --- a/src/jaqmc_legacy/pp/quadrature.py +++ b/src/jaqmc_legacy/pp/quadrature.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax.numpy as jnp diff --git a/src/jaqmc_legacy/pp/special.py b/src/jaqmc_legacy/pp/special.py index 6f97c35..8827a56 100644 --- a/src/jaqmc_legacy/pp/special.py +++ b/src/jaqmc_legacy/pp/special.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax.numpy as jnp diff --git a/src/jaqmc_legacy/pp/utils/__init__.py b/src/jaqmc_legacy/pp/utils/__init__.py index eaffc87..7f84e52 100644 --- a/src/jaqmc_legacy/pp/utils/__init__.py +++ b/src/jaqmc_legacy/pp/utils/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/src/jaqmc_legacy/pp/utils/init_electrons.py b/src/jaqmc_legacy/pp/utils/init_electrons.py index 64894da..7f504cd 100644 --- a/src/jaqmc_legacy/pp/utils/init_electrons.py +++ b/src/jaqmc_legacy/pp/utils/init_electrons.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Sequence, Tuple diff --git a/tests/__init__.py b/tests/__init__.py index b3abc0f..2250c7d 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/app/hall/hall_test.py b/tests/app/hall/hall_test.py index 0411361..296d6b6 100644 --- a/tests/app/hall/hall_test.py +++ b/tests/app/hall/hall_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for the quantum Hall workflow components.""" diff --git a/tests/app/hall/one_rdm_test.py b/tests/app/hall/one_rdm_test.py index 209adbd..df26514 100644 --- a/tests/app/hall/one_rdm_test.py +++ b/tests/app/hall/one_rdm_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for the one-body reduced density matrix (1-RDM) estimator.""" diff --git a/tests/cli_dry_run_test.py b/tests/cli_dry_run_test.py index 446e34d..2c6385f 100644 --- a/tests/cli_dry_run_test.py +++ b/tests/cli_dry_run_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import shlex diff --git a/tests/conftest.py b/tests/conftest.py index ec5fb28..9b546b0 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import multiprocessing diff --git a/tests/estimator/density_test.py b/tests/estimator/density_test.py index a1aec6c..e858b9e 100644 --- a/tests/estimator/density_test.py +++ b/tests/estimator/density_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for density histogram estimators and Kahan summation.""" diff --git a/tests/estimator/ecp/ecp_config_test.py b/tests/estimator/ecp/ecp_config_test.py index 23c293e..9b7b69c 100644 --- a/tests/estimator/ecp/ecp_config_test.py +++ b/tests/estimator/ecp/ecp_config_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for ECP configuration utilities and solid config factories. diff --git a/tests/estimator/ecp/ecp_nonlocal_test.py b/tests/estimator/ecp/ecp_nonlocal_test.py index 66f9a3a..f5b8644 100644 --- a/tests/estimator/ecp/ecp_nonlocal_test.py +++ b/tests/estimator/ecp/ecp_nonlocal_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for ECP nonlocal integral Legendre polynomial indexing. diff --git a/tests/estimator/ecp/ecp_potential_test.py b/tests/estimator/ecp/ecp_potential_test.py index 05eda93..cfca506 100644 --- a/tests/estimator/ecp/ecp_potential_test.py +++ b/tests/estimator/ecp/ecp_potential_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 r"""Tests for ECP potential evaluation and nearest-core selection. diff --git a/tests/estimator/ecp/quadrature_test.py b/tests/estimator/ecp/quadrature_test.py index e3b3670..bd6d73f 100644 --- a/tests/estimator/ecp/quadrature_test.py +++ b/tests/estimator/ecp/quadrature_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for spherical quadrature rules. diff --git a/tests/estimator/estimator_test.py b/tests/estimator/estimator_test.py index d17283b..e4b7798 100644 --- a/tests/estimator/estimator_test.py +++ b/tests/estimator/estimator_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from operator import itemgetter diff --git a/tests/estimator/ewald_test.py b/tests/estimator/ewald_test.py index 8951b2b..4b0b72e 100644 --- a/tests/estimator/ewald_test.py +++ b/tests/estimator/ewald_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax.numpy as jnp diff --git a/tests/estimator/kinetic_forward_laplacian_test.py b/tests/estimator/kinetic_forward_laplacian_test.py index e1c31c1..44ae0a1 100644 --- a/tests/estimator/kinetic_forward_laplacian_test.py +++ b/tests/estimator/kinetic_forward_laplacian_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import copy diff --git a/tests/estimator/loss_grad_test.py b/tests/estimator/loss_grad_test.py index 197066e..3b14bd8 100644 --- a/tests/estimator/loss_grad_test.py +++ b/tests/estimator/loss_grad_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/estimator/ph/ph_estimator_parity_test.py b/tests/estimator/ph/ph_estimator_parity_test.py index 1ca212b..1c3ae6d 100644 --- a/tests/estimator/ph/ph_estimator_parity_test.py +++ b/tests/estimator/ph/ph_estimator_parity_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """End-to-end correctness tests for ``PHEnergy``. diff --git a/tests/estimator/ph/ph_mixed_test.py b/tests/estimator/ph/ph_mixed_test.py index 25a0ea8..b1c00ce 100644 --- a/tests/estimator/ph/ph_mixed_test.py +++ b/tests/estimator/ph/ph_mixed_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from __future__ import annotations diff --git a/tests/estimator/ph/ph_standard_test.py b/tests/estimator/ph/ph_standard_test.py index 0f7b946..f27392e 100644 --- a/tests/estimator/ph/ph_standard_test.py +++ b/tests/estimator/ph/ph_standard_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Direct math tests for the ``standard`` PH derivative backend helpers.""" diff --git a/tests/estimator/ph/ph_workflow_test.py b/tests/estimator/ph/ph_workflow_test.py index 22a5ba0..ed1ab9b 100644 --- a/tests/estimator/ph/ph_workflow_test.py +++ b/tests/estimator/ph/ph_workflow_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from __future__ import annotations diff --git a/tests/estimator/spin_test.py b/tests/estimator/spin_test.py index 49d0074..7a5283e 100644 --- a/tests/estimator/spin_test.py +++ b/tests/estimator/spin_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for the S^2 estimator using mock wavefunctions with known values. diff --git a/tests/geometry/geometry_test.py b/tests/geometry/geometry_test.py index 084bbd8..ee9e2cc 100644 --- a/tests/geometry/geometry_test.py +++ b/tests/geometry/geometry_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for geometry modules using mathematical properties as oracles. diff --git a/tests/geometry/pbc_test.py b/tests/geometry/pbc_test.py index 2d82ffd..afa8de3 100644 --- a/tests/geometry/pbc_test.py +++ b/tests/geometry/pbc_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax.numpy as jnp diff --git a/tests/hydrogen/atom_test.py b/tests/hydrogen/atom_test.py index 2e6d82c..7a386ad 100644 --- a/tests/hydrogen/atom_test.py +++ b/tests/hydrogen/atom_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import h5py diff --git a/tests/hydrogen/molecule_test.py b/tests/hydrogen/molecule_test.py index 57bf167..0e4f473 100644 --- a/tests/hydrogen/molecule_test.py +++ b/tests/hydrogen/molecule_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import h5py diff --git a/tests/integration/__init__.py b/tests/integration/__init__.py index b3abc0f..2250c7d 100644 --- a/tests/integration/__init__.py +++ b/tests/integration/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/integration/checkpoint_test.py b/tests/integration/checkpoint_test.py index ff1b908..d1afde4 100644 --- a/tests/integration/checkpoint_test.py +++ b/tests/integration/checkpoint_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from pathlib import Path diff --git a/tests/integration/distributed_test.py b/tests/integration/distributed_test.py index 86f36d3..6eae280 100644 --- a/tests/integration/distributed_test.py +++ b/tests/integration/distributed_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Distributed multi-process training tests.""" diff --git a/tests/legacy/branching_test.py b/tests/legacy/branching_test.py index 4b3e3ec..58f1c60 100644 --- a/tests/legacy/branching_test.py +++ b/tests/legacy/branching_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import time diff --git a/tests/legacy/dmc_test.py b/tests/legacy/dmc_test.py index ebeb534..2b54ea2 100644 --- a/tests/legacy/dmc_test.py +++ b/tests/legacy/dmc_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import chex diff --git a/tests/legacy/energy_estimator_test.py b/tests/legacy/energy_estimator_test.py index a1a09a8..2d6e4e4 100644 --- a/tests/legacy/energy_estimator_test.py +++ b/tests/legacy/energy_estimator_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/legacy/position_update_test.py b/tests/legacy/position_update_test.py index 9f2eb3d..4d44078 100644 --- a/tests/legacy/position_update_test.py +++ b/tests/legacy/position_update_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import chex diff --git a/tests/legacy/recovery_test.py b/tests/legacy/recovery_test.py index 2824494..56087fc 100644 --- a/tests/legacy/recovery_test.py +++ b/tests/legacy/recovery_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import pathlib diff --git a/tests/legacy/state_io_test.py b/tests/legacy/state_io_test.py index 416a368..467bc1a 100644 --- a/tests/legacy/state_io_test.py +++ b/tests/legacy/state_io_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import os diff --git a/tests/legacy/utils_test.py b/tests/legacy/utils_test.py index 2071c6f..6a295a4 100644 --- a/tests/legacy/utils_test.py +++ b/tests/legacy/utils_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/optimizer/jasr_test.py b/tests/optimizer/jasr_test.py index 1344b71..fc384fa 100644 --- a/tests/optimizer/jasr_test.py +++ b/tests/optimizer/jasr_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from functools import partial diff --git a/tests/optimizer/kfac_test.py b/tests/optimizer/kfac_test.py index 43ee13d..3e519e0 100644 --- a/tests/optimizer/kfac_test.py +++ b/tests/optimizer/kfac_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import flax.linen as nn diff --git a/tests/utils/__init__.py b/tests/utils/__init__.py index b3abc0f..2250c7d 100644 --- a/tests/utils/__init__.py +++ b/tests/utils/__init__.py @@ -1,2 +1,2 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 diff --git a/tests/utils/atomic/factory_test.py b/tests/utils/atomic/factory_test.py index 9632b37..3de8919 100644 --- a/tests/utils/atomic/factory_test.py +++ b/tests/utils/atomic/factory_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/utils/atomic/gto_test.py b/tests/utils/atomic/gto_test.py index 5bbbe62..947cd34 100644 --- a/tests/utils/atomic/gto_test.py +++ b/tests/utils/atomic/gto_test.py @@ -1,5 +1,5 @@ # Copyright 2023 DeepMind Technologies Limited. All Rights Reserved. -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates. +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 # # This file has been modified by ByteDance Ltd. and/or its affiliates. on 2025. diff --git a/tests/utils/atomic/pp_test.py b/tests/utils/atomic/pp_test.py index ece66a8..d63e307 100644 --- a/tests/utils/atomic/pp_test.py +++ b/tests/utils/atomic/pp_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/utils/atomic/pretrain_test.py b/tests/utils/atomic/pretrain_test.py index 4b292b3..b8fc437 100644 --- a/tests/utils/atomic/pretrain_test.py +++ b/tests/utils/atomic/pretrain_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for pretraining functionality. diff --git a/tests/utils/atomic/scf_test.py b/tests/utils/atomic/scf_test.py index 6147bd6..e0d921b 100644 --- a/tests/utils/atomic/scf_test.py +++ b/tests/utils/atomic/scf_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for SCF classes (MolecularSCF and PeriodicSCF).""" diff --git a/tests/utils/chunked_vmap_test.py b/tests/utils/chunked_vmap_test.py index bebf457..ec8ee90 100644 --- a/tests/utils/chunked_vmap_test.py +++ b/tests/utils/chunked_vmap_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import chex diff --git a/tests/utils/config_test.py b/tests/utils/config_test.py index 5c34c38..cdab07a 100644 --- a/tests/utils/config_test.py +++ b/tests/utils/config_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import dataclass, field diff --git a/tests/utils/distributed.py b/tests/utils/distributed.py index 47b6c73..215cce6 100644 --- a/tests/utils/distributed.py +++ b/tests/utils/distributed.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import socket diff --git a/tests/utils/module_resolver_test.py b/tests/utils/module_resolver_test.py index f51dd5f..7ddf68d 100644 --- a/tests/utils/module_resolver_test.py +++ b/tests/utils/module_resolver_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import sys diff --git a/tests/utils/parallel_jax_test.py b/tests/utils/parallel_jax_test.py index f7da573..c4c9ad6 100644 --- a/tests/utils/parallel_jax_test.py +++ b/tests/utils/parallel_jax_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for parallel JAX utilities, especially all_gather. diff --git a/tests/utils/runtime_test.py b/tests/utils/runtime_test.py index 6ed253f..85b8ced 100644 --- a/tests/utils/runtime_test.py +++ b/tests/utils/runtime_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from unittest.mock import call diff --git a/tests/utils/wiring_test.py b/tests/utils/wiring_test.py index c47269a..961ff19 100644 --- a/tests/utils/wiring_test.py +++ b/tests/utils/wiring_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from dataclasses import dataclass diff --git a/tests/utils/yaml_format_test.py b/tests/utils/yaml_format_test.py index f48c712..6a4ebdb 100644 --- a/tests/utils/yaml_format_test.py +++ b/tests/utils/yaml_format_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import yaml diff --git a/tests/wavefunction/envelope_test.py b/tests/wavefunction/envelope_test.py index 7a3d2f5..20ccc9d 100644 --- a/tests/wavefunction/envelope_test.py +++ b/tests/wavefunction/envelope_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for the envelope system.""" diff --git a/tests/wavefunction/jastrow_test.py b/tests/wavefunction/jastrow_test.py index d0c165d..1da5cd8 100644 --- a/tests/wavefunction/jastrow_test.py +++ b/tests/wavefunction/jastrow_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for Jastrow factor implementation.""" diff --git a/tests/wavefunction/molecule_wavefunction_test.py b/tests/wavefunction/molecule_wavefunction_test.py index f5d49fc..08d0604 100644 --- a/tests/wavefunction/molecule_wavefunction_test.py +++ b/tests/wavefunction/molecule_wavefunction_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 """Tests for molecular wavefunction implementations. diff --git a/tests/wavefunction/solid_test.py b/tests/wavefunction/solid_test.py index 9a5f78a..dd5f144 100644 --- a/tests/wavefunction/solid_test.py +++ b/tests/wavefunction/solid_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import jax diff --git a/tests/workflow/preset_test.py b/tests/workflow/preset_test.py index 66896a0..ef22b12 100644 --- a/tests/workflow/preset_test.py +++ b/tests/workflow/preset_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from typing import Any diff --git a/tests/writer/console_writer_test.py b/tests/writer/console_writer_test.py index bdf2aaf..981b784 100644 --- a/tests/writer/console_writer_test.py +++ b/tests/writer/console_writer_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 from jaqmc.writer.console import ConsoleWriter diff --git a/tests/writer/writer_test.py b/tests/writer/writer_test.py index 7b9dc01..f675d96 100644 --- a/tests/writer/writer_test.py +++ b/tests/writer/writer_test.py @@ -1,4 +1,4 @@ -# Copyright (c) 2025-2026 Bytedance Ltd. and/or its affiliates +# Copyright (c) 2025-2026 ByteDance Ltd. and/or its affiliates # SPDX-License-Identifier: Apache-2.0 import sys