LICENSE
MANIFEST.in
NOTICE
README.md
setup.py
src/agent/__init__.py
src/agent/agent/__init__.py
src/agent/agent/cot_agent.py
src/agent/agent/onescience_agent.py
src/agent/agent/react_agent.py
src/agent/agent/unit/__init__.py
src/agent/agent/unit/agent_state.py
src/agent/agent/unit/cot_runner.py
src/agent/agent/unit/generate_runner.py
src/agent/agent/unit/runner.py
src/agent/agent/unit/task_classify_runner.py
src/agent/agent/unit/tool_manager.py
src/agent/agent/unit/tool_runner.py
src/agent/llm/__init__.py
src/agent/llm/chat_model.py
src/agent/llm/embedding_model.py
src/agent/llm/rerank_model.py
src/agent/rag/__init__.py
src/agent/rag/rag_server.py
src/agent/rag/chunks/__init__.py
src/agent/rag/chunks/fixed_text_splitter.py
src/agent/rag/datasource/__init__.py
src/agent/rag/datasource/retrieval_methods.py
src/agent/rag/datasource/retrieval_service.py
src/agent/rag/datasource/keyword/__init__.py
src/agent/rag/datasource/keyword/keyword_base.py
src/agent/rag/datasource/keyword/keyword_factory.py
src/agent/rag/datasource/keyword/jieba/__init__.py
src/agent/rag/datasource/keyword/jieba/jieba_keyword_table_handler.py
src/agent/rag/datasource/keyword/jieba/stopwords.py
src/agent/rag/datasource/vdb/__init__.py
src/agent/rag/datasource/vdb/constant.py
src/agent/rag/datasource/vdb/vector_base.py
src/agent/rag/datasource/vdb/vector_factory.py
src/agent/rag/datasource/vdb/milvus/__init__.py
src/agent/rag/datasource/vdb/milvus/milvus_vector.py
src/agent/rag/datasource/vdb/milvus/schemas.py
src/agent/rag/docparser/__init__.py
src/agent/rag/docparser/doc_processor.py
src/agent/rag/docparser/helpers.py
src/agent/rag/docparser/markdown_parser.py
src/agent/rag/docparser/text_parser.py
src/agent/rag/postprocess/__init__.py
src/agent/rag/postprocess/rerank_base.py
src/agent/rag/postprocess/rerank_factory.py
src/agent/rag/postprocess/rerank_model.py
src/agent/rag/postprocess/weight_rerank.py
src/agent/rag/preprocess/__init__.py
src/agent/rag/preprocess/collection_preprocessor.py
src/agent/rag/preprocess/summary_preprocessor.py
src/agent/untils/__init__.py
src/agent/untils/until.py
src/onescience/__init__.py
src/onescience/constants.py
src/onescience.egg-info/PKG-INFO
src/onescience.egg-info/SOURCES.txt
src/onescience.egg-info/dependency_links.txt
src/onescience.egg-info/not-zip-safe
src/onescience.egg-info/requires.txt
src/onescience.egg-info/top_level.txt
src/onescience/datapipes/__init__.py
src/onescience/datapipes/datapipe.py
src/onescience/datapipes/meta.py
src/onescience/datapipes/cae/__init__.py
src/onescience/datapipes/cae/readers.py
src/onescience/datapipes/cfd_benchmark/__init__.py
src/onescience/datapipes/cfd_benchmark/data_factory.py
src/onescience/datapipes/cfd_benchmark/data_loader.py
src/onescience/datapipes/cfd_benchmark/shapenet_utils.py
src/onescience/datapipes/climate/__init__.py
src/onescience/datapipes/climate/era5_hdf5.py
src/onescience/datapipes/climate/synthetic.py
src/onescience/datapipes/climate/utils/__init__.py
src/onescience/datapipes/climate/utils/invariant.py
src/onescience/datapipes/climate/utils/zenith_angle.py
src/onescience/datapipes/eagle/__init__.py
src/onescience/datapipes/eagle/eagle.py
src/onescience/datapipes/gnn/__init__.py
src/onescience/datapipes/gnn/ahmed_body_dataset.py
src/onescience/datapipes/gnn/bsms.py
src/onescience/datapipes/gnn/drivaernet_dataset.py
src/onescience/datapipes/gnn/lagrangian_dataset.py
src/onescience/datapipes/gnn/lagrangian_reading_utils.py
src/onescience/datapipes/gnn/stokes_dataset.py
src/onescience/datapipes/gnn/utils.py
src/onescience/datapipes/gnn/vortex_shedding_dataset.py
src/onescience/datapipes/gnn/vortex_shedding_re300_1000_dataset.py
src/onescience/datapipes/healpix/__init__.py
src/onescience/datapipes/healpix/coupledtimeseries_dataset.py
src/onescience/datapipes/healpix/couplers.py
src/onescience/datapipes/healpix/data_modules.py
src/onescience/datapipes/healpix/timeseries_dataset.py
src/onescience/datapipes/openfold/__init__.py
src/onescience/datapipes/openfold/data_transforms.py
src/onescience/datapipes/openfold/errors.py
src/onescience/datapipes/openfold/mmcif_parsing.py
src/onescience/datapipes/openfold/msa_identifiers.py
src/onescience/datapipes/openfold/msa_pairing.py
src/onescience/datapipes/openfold/parsers.py
src/onescience/datapipes/openfold/templates.py
src/onescience/datapipes/openfold/tools/__init__.py
src/onescience/datapipes/openfold/tools/jackhmmer.py
src/onescience/datapipes/openfold/tools/utils.py
src/onescience/datapipes/protenix/__init__.py
src/onescience/datapipes/protenix/ccd.py
src/onescience/datapipes/protenix/constants.py
src/onescience/datapipes/protenix/data_pipeline.py
src/onescience/datapipes/protenix/dataloader.py
src/onescience/datapipes/protenix/dataset.py
src/onescience/datapipes/protenix/featurizer.py
src/onescience/datapipes/protenix/filter.py
src/onescience/datapipes/protenix/infer_data_pipeline.py
src/onescience/datapipes/protenix/json_maker.py
src/onescience/datapipes/protenix/json_parser.py
src/onescience/datapipes/protenix/json_to_feature.py
src/onescience/datapipes/protenix/msa_featurizer.py
src/onescience/datapipes/protenix/msa_utils.py
src/onescience/datapipes/protenix/parser.py
src/onescience/datapipes/protenix/substructure_perms.py
src/onescience/datapipes/protenix/tokenizer.py
src/onescience/datapipes/protenix/utils.py
src/onescience/datapipes/transolver/__init__.py
src/onescience/datapipes/transolver/dataset.py
src/onescience/datapipes/transolver/load_dataset.py
src/onescience/datapipes/uma/__init__.py
src/onescience/datapipes/uma/_utils.py
src/onescience/datapipes/uma/ase_datasets.py
src/onescience/datapipes/uma/atomic_data.py
src/onescience/datapipes/uma/base_dataset.py
src/onescience/datapipes/uma/mt_concat_dataset.py
src/onescience/datapipes/uma/target_metadata_guesser.py
src/onescience/datapipes/uma/collaters/__init__.py
src/onescience/datapipes/uma/collaters/mt_collater.py
src/onescience/datapipes/uma/collaters/simple_collater.py
src/onescience/datapipes/uma/embeddings/__init__.py
src/onescience/datapipes/uma/embeddings/atomic_radii.py
src/onescience/datapipes/uma/embeddings/continuous_embeddings.py
src/onescience/datapipes/uma/embeddings/khot_embeddings.py
src/onescience/datapipes/uma/embeddings/qmof_khot_embeddings.py
src/onescience/datapipes/uma/samplers/__init__.py
src/onescience/datapipes/uma/samplers/max_atom_distributed_sampler.py
src/onescience/datapipes/votex/__init__.py
src/onescience/datapipes/votex/ahmed_body_dataset.py
src/onescience/datapipes/votex/bsms.py
src/onescience/datapipes/votex/drivaernet_dataset.py
src/onescience/datapipes/votex/stokes_dataset.py
src/onescience/datapipes/votex/utils.py
src/onescience/datapipes/votex/vortex_shedding_dataset.py
src/onescience/datapipes/votex/vortex_shedding_re300_1000_dataset.py
src/onescience/deploy/__init__.py
src/onescience/deploy/onnx/__init__.py
src/onescience/deploy/onnx/utils.py
src/onescience/deploy/triton/__init__.py
src/onescience/distributed/__init__.py
src/onescience/distributed/autograd.py
src/onescience/distributed/comm.py
src/onescience/distributed/config.py
src/onescience/distributed/dualpipeutils.py
src/onescience/distributed/dualpipev.py
src/onescience/distributed/fft.py
src/onescience/distributed/manager.py
src/onescience/distributed/mappings.py
src/onescience/distributed/utils.py
src/onescience/flax_models/__init__.py
src/onescience/flax_models/MolSculptor/README.md
src/onescience/flax_models/MolSculptor/__init__.py
src/onescience/flax_models/MolSculptor/configs.py
src/onescience/flax_models/MolSculptor/package_config.py
src/onescience/flax_models/MolSculptor/utils.py
src/onescience/flax_models/MolSculptor/checkpoints/__init__.py
src/onescience/flax_models/MolSculptor/checkpoints/auto-encoder/__init__.py
src/onescience/flax_models/MolSculptor/checkpoints/auto-encoder/config.pkl
src/onescience/flax_models/MolSculptor/checkpoints/diffusion-transformer/__init__.py
src/onescience/flax_models/MolSculptor/checkpoints/diffusion-transformer/config_denovo.pkl
src/onescience/flax_models/MolSculptor/checkpoints/diffusion-transformer/config_opt.pkl
src/onescience/flax_models/MolSculptor/dsdp/make_dsdp.txt
src/onescience/flax_models/MolSculptor/net/__init__.py
src/onescience/flax_models/MolSculptor/net/decoder.py
src/onescience/flax_models/MolSculptor/net/embedding.py
src/onescience/flax_models/MolSculptor/net/encoder.py
src/onescience/flax_models/MolSculptor/net/generator.py
src/onescience/flax_models/MolSculptor/net/molct.py
src/onescience/flax_models/MolSculptor/net/molct_plus.py
src/onescience/flax_models/MolSculptor/net/quant.py
src/onescience/flax_models/MolSculptor/src/__init__.py
src/onescience/flax_models/MolSculptor/src/common/__init__.py
src/onescience/flax_models/MolSculptor/src/common/config.py
src/onescience/flax_models/MolSculptor/src/common/train.py
src/onescience/flax_models/MolSculptor/src/common/utils.py
src/onescience/flax_models/MolSculptor/src/common/layers/__init__.py
src/onescience/flax_models/MolSculptor/src/common/layers/dense.py
src/onescience/flax_models/MolSculptor/src/common/layers/mlp.py
src/onescience/flax_models/MolSculptor/src/common/pallas/__init__.py
src/onescience/flax_models/MolSculptor/src/common/pallas/attention.py
src/onescience/flax_models/MolSculptor/src/common/pallas/attention_withbias.py
src/onescience/flax_models/MolSculptor/src/common/rbf/__init__.py
src/onescience/flax_models/MolSculptor/src/common/rbf/bessel.py
src/onescience/flax_models/MolSculptor/src/common/rbf/gaussian.py
src/onescience/flax_models/MolSculptor/src/common/rbf/loggaussian.py
src/onescience/flax_models/MolSculptor/src/model/__init__.py
src/onescience/flax_models/MolSculptor/src/model/diffusion_transformer.py
src/onescience/flax_models/MolSculptor/src/model/example_config.py
src/onescience/flax_models/MolSculptor/src/model/qformer.py
src/onescience/flax_models/MolSculptor/src/model/transformer.py
src/onescience/flax_models/MolSculptor/src/module/__init__.py
src/onescience/flax_models/MolSculptor/src/module/attention.py
src/onescience/flax_models/MolSculptor/src/module/transformer.py
src/onescience/flax_models/MolSculptor/train/__init__.py
src/onescience/flax_models/MolSculptor/train/dataloader.py
src/onescience/flax_models/MolSculptor/train/inference.py
src/onescience/flax_models/MolSculptor/train/rewards.py
src/onescience/flax_models/MolSculptor/train/scheduler.py
src/onescience/flax_models/MolSculptor/train/utils.py
src/onescience/flax_models/MolSculptor/train/withloss.py
src/onescience/flax_models/Pt_DiT/__init__.py
src/onescience/flax_models/Pt_DiT/common/__init__.py
src/onescience/flax_models/Pt_DiT/common/config.py
src/onescience/flax_models/Pt_DiT/common/train.py
src/onescience/flax_models/Pt_DiT/common/utils.py
src/onescience/flax_models/Pt_DiT/common/layers/__init__.py
src/onescience/flax_models/Pt_DiT/common/layers/dense.py
src/onescience/flax_models/Pt_DiT/common/pallas/__init__.py
src/onescience/flax_models/Pt_DiT/common/pallas/attention.py
src/onescience/flax_models/Pt_DiT/common/pallas/attention_withbias.py
src/onescience/flax_models/Pt_DiT/common/rbf/__init__.py
src/onescience/flax_models/Pt_DiT/common/rbf/bessel.py
src/onescience/flax_models/Pt_DiT/common/rbf/gaussian.py
src/onescience/flax_models/Pt_DiT/common/rbf/loggaussian.py
src/onescience/flax_models/Pt_DiT/model/__init__.py
src/onescience/flax_models/Pt_DiT/model/diffusion_transformer.py
src/onescience/flax_models/Pt_DiT/model/example_config.py
src/onescience/flax_models/Pt_DiT/model/qformer.py
src/onescience/flax_models/Pt_DiT/model/transformer.py
src/onescience/flax_models/Pt_DiT/module/__init__.py
src/onescience/flax_models/Pt_DiT/module/attention.py
src/onescience/flax_models/Pt_DiT/module/transformer.py
src/onescience/flax_models/Pt_DiT/train/__init__.py
src/onescience/flax_models/Pt_DiT/train/schedulers.py
src/onescience/flax_models/alphafold3/README.md
src/onescience/flax_models/alphafold3/__init__.py
src/onescience/flax_models/alphafold3/build_data.py
src/onescience/flax_models/alphafold3/build_extension.py
src/onescience/flax_models/alphafold3/package_config.py
src/onescience/flax_models/alphafold3/version.py
src/onescience/flax_models/alphafold3/common/__init__.py
src/onescience/flax_models/alphafold3/common/base_config.py
src/onescience/flax_models/alphafold3/common/folding_input.py
src/onescience/flax_models/alphafold3/common/resources.py
src/onescience/flax_models/alphafold3/common/testing/__init__.py
src/onescience/flax_models/alphafold3/common/testing/data.py
src/onescience/flax_models/alphafold3/constants/__init__.py
src/onescience/flax_models/alphafold3/constants/atom_types.py
src/onescience/flax_models/alphafold3/constants/chemical_component_sets.py
src/onescience/flax_models/alphafold3/constants/chemical_components.py
src/onescience/flax_models/alphafold3/constants/mmcif_names.py
src/onescience/flax_models/alphafold3/constants/periodic_table.py
src/onescience/flax_models/alphafold3/constants/residue_names.py
src/onescience/flax_models/alphafold3/constants/side_chains.py
src/onescience/flax_models/alphafold3/constants/converters/__init__.py
src/onescience/flax_models/alphafold3/constants/converters/ccd_pickle_gen.py
src/onescience/flax_models/alphafold3/constants/converters/chemical_component_sets_gen.py
src/onescience/flax_models/alphafold3/data/__init__.py
src/onescience/flax_models/alphafold3/data/featurisation.py
src/onescience/flax_models/alphafold3/data/msa.py
src/onescience/flax_models/alphafold3/data/msa_config.py
src/onescience/flax_models/alphafold3/data/msa_features.py
src/onescience/flax_models/alphafold3/data/msa_identifiers.py
src/onescience/flax_models/alphafold3/data/msa_store.py
src/onescience/flax_models/alphafold3/data/parsers.py
src/onescience/flax_models/alphafold3/data/pipeline.py
src/onescience/flax_models/alphafold3/data/structure_stores.py
src/onescience/flax_models/alphafold3/data/template_realign.py
src/onescience/flax_models/alphafold3/data/template_store.py
src/onescience/flax_models/alphafold3/data/templates.py
src/onescience/flax_models/alphafold3/data/cpp/__init__.py
src/onescience/flax_models/alphafold3/data/tools/__init__.py
src/onescience/flax_models/alphafold3/data/tools/hmmalign.py
src/onescience/flax_models/alphafold3/data/tools/hmmbuild.py
src/onescience/flax_models/alphafold3/data/tools/hmmsearch.py
src/onescience/flax_models/alphafold3/data/tools/jackhmmer.py
src/onescience/flax_models/alphafold3/data/tools/mmseqs.py
src/onescience/flax_models/alphafold3/data/tools/msa_tool.py
src/onescience/flax_models/alphafold3/data/tools/nhmmer.py
src/onescience/flax_models/alphafold3/data/tools/rdkit_utils.py
src/onescience/flax_models/alphafold3/data/tools/subprocess_utils.py
src/onescience/flax_models/alphafold3/jax/__init__.py
src/onescience/flax_models/alphafold3/jax/attention/__init__.py
src/onescience/flax_models/alphafold3/jax/attention/attention.py
src/onescience/flax_models/alphafold3/jax/attention/attention_base.py
src/onescience/flax_models/alphafold3/jax/attention/attention_call_arg_specs.py
src/onescience/flax_models/alphafold3/jax/attention/flash_attention.py
src/onescience/flax_models/alphafold3/jax/attention/xla_attention.py
src/onescience/flax_models/alphafold3/jax/common/__init__.py
src/onescience/flax_models/alphafold3/jax/common/array_view.py
src/onescience/flax_models/alphafold3/jax/common/precision.py
src/onescience/flax_models/alphafold3/jax/common/triton_utils.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/__init__.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/block.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/gated_linear_unit.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/gated_linear_unit_base.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/matmul_config.py
src/onescience/flax_models/alphafold3/jax/gated_linear_unit/matmul_ext.py
src/onescience/flax_models/alphafold3/jax/geometry/__init__.py
src/onescience/flax_models/alphafold3/jax/geometry/rigid_matrix_vector.py
src/onescience/flax_models/alphafold3/jax/geometry/rotation_matrix.py
src/onescience/flax_models/alphafold3/jax/geometry/struct_of_array.py
src/onescience/flax_models/alphafold3/jax/geometry/utils.py
src/onescience/flax_models/alphafold3/jax/geometry/vector.py
src/onescience/flax_models/alphafold3/model/__init__.py
src/onescience/flax_models/alphafold3/model/confidence_types.py
src/onescience/flax_models/alphafold3/model/confidences.py
src/onescience/flax_models/alphafold3/model/data3.py
src/onescience/flax_models/alphafold3/model/data_constants.py
src/onescience/flax_models/alphafold3/model/feat_batch.py
src/onescience/flax_models/alphafold3/model/features.py
src/onescience/flax_models/alphafold3/model/merging_features.py
src/onescience/flax_models/alphafold3/model/mmcif_metadata.py
src/onescience/flax_models/alphafold3/model/model.py
src/onescience/flax_models/alphafold3/model/model_config.py
src/onescience/flax_models/alphafold3/model/msa_pairing.py
src/onescience/flax_models/alphafold3/model/params.py
src/onescience/flax_models/alphafold3/model/post_processing.py
src/onescience/flax_models/alphafold3/model/protein_data_processing.py
src/onescience/flax_models/alphafold3/model/atom_layout/__init__.py
src/onescience/flax_models/alphafold3/model/atom_layout/atom_layout.py
src/onescience/flax_models/alphafold3/model/components/__init__.py
src/onescience/flax_models/alphafold3/model/components/haiku_modules.py
src/onescience/flax_models/alphafold3/model/components/mapping.py
src/onescience/flax_models/alphafold3/model/components/utils.py
src/onescience/flax_models/alphafold3/model/network/__init__.py
src/onescience/flax_models/alphafold3/model/network/atom_cross_attention.py
src/onescience/flax_models/alphafold3/model/network/confidence_head.py
src/onescience/flax_models/alphafold3/model/network/diffusion_head.py
src/onescience/flax_models/alphafold3/model/network/diffusion_transformer.py
src/onescience/flax_models/alphafold3/model/network/distogram_head.py
src/onescience/flax_models/alphafold3/model/network/evoformer.py
src/onescience/flax_models/alphafold3/model/network/featurization.py
src/onescience/flax_models/alphafold3/model/network/modules.py
src/onescience/flax_models/alphafold3/model/network/noise_level_embeddings.py
src/onescience/flax_models/alphafold3/model/network/template_modules.py
src/onescience/flax_models/alphafold3/model/pipeline/__init__.py
src/onescience/flax_models/alphafold3/model/pipeline/inter_chain_bonds.py
src/onescience/flax_models/alphafold3/model/pipeline/pipeline.py
src/onescience/flax_models/alphafold3/model/pipeline/structure_cleaning.py
src/onescience/flax_models/alphafold3/model/scoring/__init__.py
src/onescience/flax_models/alphafold3/model/scoring/alignment.py
src/onescience/flax_models/alphafold3/model/scoring/chirality.py
src/onescience/flax_models/alphafold3/model/scoring/covalent_bond_cleaning.py
src/onescience/flax_models/alphafold3/model/scoring/scoring.py
src/onescience/flax_models/alphafold3/parsers/__init__.py
src/onescience/flax_models/alphafold3/parsers/cpp/__init__.py
src/onescience/flax_models/alphafold3/parsers/cpp/cif_dict.pyi
src/onescience/flax_models/alphafold3/parsers/cpp/fasta_iterator.pyi
src/onescience/flax_models/alphafold3/parsers/cpp/msa_conversion.pyi
src/onescience/flax_models/alphafold3/scripts/__init__.py
src/onescience/flax_models/alphafold3/structure/__init__.py
src/onescience/flax_models/alphafold3/structure/bioassemblies.py
src/onescience/flax_models/alphafold3/structure/bonds.py
src/onescience/flax_models/alphafold3/structure/chemical_components.py
src/onescience/flax_models/alphafold3/structure/mmcif.py
src/onescience/flax_models/alphafold3/structure/parsing.py
src/onescience/flax_models/alphafold3/structure/sterics.py
src/onescience/flax_models/alphafold3/structure/structure.py
src/onescience/flax_models/alphafold3/structure/structure_tables.py
src/onescience/flax_models/alphafold3/structure/table.py
src/onescience/flax_models/alphafold3/structure/test_utils.py
src/onescience/flax_models/alphafold3/structure/cpp/__init__.py
src/onescience/flax_models/alphafold3/structure/cpp/aggregation.pyi
src/onescience/flax_models/alphafold3/structure/cpp/membership.pyi
src/onescience/flax_models/alphafold3/structure/cpp/mmcif_atom_site.pyi
src/onescience/flax_models/alphafold3/structure/cpp/mmcif_layout.pyi
src/onescience/flax_models/alphafold3/structure/cpp/mmcif_struct_conn.pyi
src/onescience/flax_models/alphafold3/structure/cpp/mmcif_utils.pyi
src/onescience/flax_models/alphafold3/structure/cpp/string_array.pyi
src/onescience/flax_models/alphafold3/test_data/__init__.py
src/onescience/flax_models/alphafold3/test_data/featurised_example.json
src/onescience/flax_models/alphafold3/test_data/model_config.json
src/onescience/flax_models/alphafold3/test_data/miniature_databases/__init__.py
src/onescience/flax_models/alphafold3/test_data/miniature_databases/bfd-first_non_consensus_sequences__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/mgy_clusters__subsampled_1000.fa
src/onescience/flax_models/alphafold3/test_data/miniature_databases/nt_rna_2023_02_23_clust_seq_id_90_cov_80_rep_seq__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_seqres_2022_09_28__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/rfam_14_4_clustered_rep_seq__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/rnacentral_active_seq_id_90_cov_80_linclust__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/uniprot_all__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/uniref90__subsampled_1000.fasta
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_mmcif/5y2e.cif
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_mmcif/6s61.cif
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_mmcif/6ydw.cif
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_mmcif/7rye.cif
src/onescience/flax_models/alphafold3/test_data/miniature_databases/pdb_mmcif/__init__.py
src/onescience/flax_models/protoken/__init__.py
src/onescience/flax_models/protoken/common/__init__.py
src/onescience/flax_models/protoken/common/config_load.py
src/onescience/flax_models/protoken/common/geometry.py
src/onescience/flax_models/protoken/common/multimer_block.py
src/onescience/flax_models/protoken/common/protein.py
src/onescience/flax_models/protoken/common/residue_constants.py
src/onescience/flax_models/protoken/common/utils.py
src/onescience/flax_models/protoken/config/__init__.py
src/onescience/flax_models/protoken/config/global_config.py
src/onescience/flax_models/protoken/config/train_vq_config.py
src/onescience/flax_models/protoken/data/__init__.py
src/onescience/flax_models/protoken/data/dataset.py
src/onescience/flax_models/protoken/data/preprocess.py
src/onescience/flax_models/protoken/data/protein_utils.py
src/onescience/flax_models/protoken/data/utils.py
src/onescience/flax_models/protoken/inference/__init__.py
src/onescience/flax_models/protoken/inference/inference.py
src/onescience/flax_models/protoken/loss/CA_distogram_loss.py
src/onescience/flax_models/protoken/loss/__init__.py
src/onescience/flax_models/protoken/loss/confidence_loss.py
src/onescience/flax_models/protoken/loss/fape_loss.py
src/onescience/flax_models/protoken/loss/inverse_folding_loss.py
src/onescience/flax_models/protoken/loss/structure_violation_loss.py
src/onescience/flax_models/protoken/loss/utils.py
src/onescience/flax_models/protoken/model/__init__.py
src/onescience/flax_models/protoken/model/bottleneck.py
src/onescience/flax_models/protoken/model/decoder.py
src/onescience/flax_models/protoken/model/encoder.py
src/onescience/flax_models/protoken/model/equivariant.py
src/onescience/flax_models/protoken/model/flash_evoformer.py
src/onescience/flax_models/protoken/model/transformers.py
src/onescience/flax_models/protoken/model/utils.py
src/onescience/flax_models/protoken/model/geometry/__init__.py
src/onescience/flax_models/protoken/model/geometry/rigid_matrix_vector.py
src/onescience/flax_models/protoken/model/geometry/rotation_matrix.py
src/onescience/flax_models/protoken/model/geometry/struct_of_array.py
src/onescience/flax_models/protoken/model/geometry/test_utils.py
src/onescience/flax_models/protoken/model/geometry/utils.py
src/onescience/flax_models/protoken/model/geometry/vector.py
src/onescience/flax_models/protoken/modules/__init__.py
src/onescience/flax_models/protoken/modules/basic.py
src/onescience/flax_models/protoken/modules/head.py
src/onescience/flax_models/protoken/modules/structure.py
src/onescience/flax_models/protoken/modules/templates.py
src/onescience/flax_models/protoken/modules/transformer_blocks.py
src/onescience/flax_models/protoken/scripts/__init__.py
src/onescience/flax_models/protoken/scripts/decode_structure.py
src/onescience/flax_models/protoken/scripts/infer.py
src/onescience/flax_models/protoken/scripts/infer_batch.py
src/onescience/flax_models/protoken/tokenizer/__init__.py
src/onescience/flax_models/protoken/tokenizer/vector_quantization.py
src/onescience/flax_models/protoken/train/__init__.py
src/onescience/flax_models/protoken/train/sharding.py
src/onescience/flax_models/protoken/train/utils.py
src/onescience/kernels/__init__.py
src/onescience/kernels/layer_norm.py
src/onescience/launch/__init__.py
src/onescience/launch/config/__init__.py
src/onescience/launch/logging/__init__.py
src/onescience/launch/logging/console.py
src/onescience/launch/logging/launch.py
src/onescience/launch/logging/mlflow.py
src/onescience/launch/logging/utils.py
src/onescience/launch/logging/wandb.py
src/onescience/launch/utils/__init__.py
src/onescience/launch/utils/checkpoint.py
src/onescience/memory/__init__.py
src/onescience/memory/checkpoint.py
src/onescience/memory/deepspeedcapture.py
src/onescience/memory/memoryhook.py
src/onescience/metrics/__init__.py
src/onescience/metrics/climate/__init__.py
src/onescience/metrics/climate/acc.py
src/onescience/metrics/climate/efi.py
src/onescience/metrics/climate/healpix_loss.py
src/onescience/metrics/climate/loss.py
src/onescience/metrics/climate/reduction.py
src/onescience/metrics/diffusion/__init__.py
src/onescience/metrics/diffusion/fid.py
src/onescience/metrics/diffusion/loss.py
src/onescience/metrics/general/__init__.py
src/onescience/metrics/general/calibration.py
src/onescience/metrics/general/crps.py
src/onescience/metrics/general/ensemble_metrics.py
src/onescience/metrics/general/entropy.py
src/onescience/metrics/general/histogram.py
src/onescience/metrics/general/mse.py
src/onescience/metrics/general/reduction.py
src/onescience/metrics/general/wasserstein.py
src/onescience/metrics/protenix/__init__.py
src/onescience/metrics/protenix/clash.py
src/onescience/metrics/protenix/lddt_metrics.py
src/onescience/metrics/protenix/rmsd.py
src/onescience/models/__init__.py
src/onescience/models/fcn_mip_plugin.py
src/onescience/models/meta.py
src/onescience/models/module.py
src/onescience/models/GPs/__init__.py
src/onescience/models/GPs/gpplus_model.py
src/onescience/models/GPs/gpregression.py
src/onescience/models/GPs/horseshoe.py
src/onescience/models/GPs/mollified_uniform.py
src/onescience/models/UMA/__init__.py
src/onescience/models/UMA/_cli.py
src/onescience/models/UMA/_config.py
src/onescience/models/UMA/calculate/__init__.py
src/onescience/models/UMA/calculate/ase_calculator.py
src/onescience/models/UMA/calculate/pretrained_mlip.py
src/onescience/models/UMA/common/__init__.py
src/onescience/models/UMA/common/data_parallel.py
src/onescience/models/UMA/common/distutils.py
src/onescience/models/UMA/common/gp_utils.py
src/onescience/models/UMA/common/logger.py
src/onescience/models/UMA/common/profiler_utils.py
src/onescience/models/UMA/common/registry.py
src/onescience/models/UMA/common/test_utils.py
src/onescience/models/UMA/common/tutorial_utils.py
src/onescience/models/UMA/common/typing.py
src/onescience/models/UMA/common/utils.py
src/onescience/models/UMA/components/__init__.py
src/onescience/models/UMA/components/reducer.py
src/onescience/models/UMA/components/runner.py
src/onescience/models/UMA/components/utils.py
src/onescience/models/UMA/components/benchmark/__init__.py
src/onescience/models/UMA/components/benchmark/adsorbml_reducer.py
src/onescience/models/UMA/components/benchmark/adsorption_reducer.py
src/onescience/models/UMA/components/benchmark/benchmark_reducer.py
src/onescience/models/UMA/components/benchmark/kappa_reducer.py
src/onescience/models/UMA/components/benchmark/materials_discovery_reducer.py
src/onescience/models/UMA/components/benchmark/nvemd_reducer.py
src/onescience/models/UMA/components/benchmark/omc_polymorph_reducer.py
src/onescience/models/UMA/components/benchmark/uma_speed_benchmark.py
src/onescience/models/UMA/components/calculate/__init__.py
src/onescience/models/UMA/components/calculate/adsorbml_runner.py
src/onescience/models/UMA/components/calculate/adsorption_runner.py
src/onescience/models/UMA/components/calculate/adsorption_singlepoint_runner.py
src/onescience/models/UMA/components/calculate/calculate_runner.py
src/onescience/models/UMA/components/calculate/elasticity_runner.py
src/onescience/models/UMA/components/calculate/kappa_runner.py
src/onescience/models/UMA/components/calculate/nve_md_runner.py
src/onescience/models/UMA/components/calculate/pairwise_ct_runner.py
src/onescience/models/UMA/components/calculate/phonon_runner.py
src/onescience/models/UMA/components/calculate/relaxation_runner.py
src/onescience/models/UMA/components/calculate/singlepoint_runner.py
src/onescience/models/UMA/components/calculate/recipes/__init__.py
src/onescience/models/UMA/components/calculate/recipes/adsorbml.py
src/onescience/models/UMA/components/calculate/recipes/adsorption.py
src/onescience/models/UMA/components/calculate/recipes/elastic.py
src/onescience/models/UMA/components/calculate/recipes/energy.py
src/onescience/models/UMA/components/calculate/recipes/local_env.py
src/onescience/models/UMA/components/calculate/recipes/phonons.py
src/onescience/models/UMA/components/calculate/recipes/relax.py
src/onescience/models/UMA/components/calculate/recipes/thermal_conductivity.py
src/onescience/models/UMA/components/calculate/recipes/utils.py
src/onescience/models/UMA/components/common/__init__.py
src/onescience/models/UMA/components/common/dataloader_builder.py
src/onescience/models/UMA/components/common/flops_profile.py
src/onescience/models/UMA/components/common/load_dataframe.py
src/onescience/models/UMA/components/evaluate/__init__.py
src/onescience/models/UMA/components/evaluate/eval_runner.py
src/onescience/models/UMA/components/train/__init__.py
src/onescience/models/UMA/components/train/train_runner.py
src/onescience/models/UMA/graph/__init__.py
src/onescience/models/UMA/graph/compute.py
src/onescience/models/UMA/graph/radius_graph_pbc.py
src/onescience/models/UMA/models/__init__.py
src/onescience/models/UMA/models/base.py
src/onescience/models/UMA/models/uma/__init__.py
src/onescience/models/UMA/models/uma/escn_md.py
src/onescience/models/UMA/models/uma/escn_md_block.py
src/onescience/models/UMA/models/uma/escn_moe.py
src/onescience/models/UMA/models/uma/common/__init__.py
src/onescience/models/UMA/models/uma/common/rotation.py
src/onescience/models/UMA/models/uma/common/rotation_cuda_graph.py
src/onescience/models/UMA/models/uma/common/sampling.py
src/onescience/models/UMA/models/uma/common/so3.py
src/onescience/models/UMA/models/uma/nn/__init__.py
src/onescience/models/UMA/models/uma/nn/activation.py
src/onescience/models/UMA/models/uma/nn/dropout.py
src/onescience/models/UMA/models/uma/nn/embedding_dev.py
src/onescience/models/UMA/models/uma/nn/layer_norm.py
src/onescience/models/UMA/models/uma/nn/mole.py
src/onescience/models/UMA/models/uma/nn/mole_utils.py
src/onescience/models/UMA/models/uma/nn/radial.py
src/onescience/models/UMA/models/uma/nn/so2_layers.py
src/onescience/models/UMA/models/uma/nn/so3_layers.py
src/onescience/models/UMA/models/utils/__init__.py
src/onescience/models/UMA/models/utils/activations.py
src/onescience/models/UMA/models/utils/basis.py
src/onescience/models/UMA/models/utils/irreps.py
src/onescience/models/UMA/modules/__init__.py
src/onescience/models/UMA/modules/evaluator.py
src/onescience/models/UMA/modules/exponential_moving_average.py
src/onescience/models/UMA/modules/loss.py
src/onescience/models/UMA/modules/scheduler.py
src/onescience/models/UMA/modules/transforms.py
src/onescience/models/UMA/modules/normalization/__init__.py
src/onescience/models/UMA/modules/normalization/_load_utils.py
src/onescience/models/UMA/modules/normalization/element_references.py
src/onescience/models/UMA/modules/normalization/normalizer.py
src/onescience/models/UMA/modules/scaling/__init__.py
src/onescience/models/UMA/modules/scaling/compat.py
src/onescience/models/UMA/modules/scaling/fit.py
src/onescience/models/UMA/modules/scaling/scale_factor.py
src/onescience/models/UMA/modules/scaling/util.py
src/onescience/models/UMA/scripts/__init__.py
src/onescience/models/UMA/scripts/convert_hydra_to_release.py
src/onescience/models/UMA/scripts/create_finetune_dataset.py
src/onescience/models/UMA/scripts/create_uma_finetune_dataset.py
src/onescience/models/UMA/scripts/download_data.py
src/onescience/models/UMA/scripts/download_large_files.py
src/onescience/models/UMA/scripts/eqv2_to_hydra_eqv2.py
src/onescience/models/UMA/scripts/fit_normalizers.py
src/onescience/models/UMA/scripts/fit_references.py
src/onescience/models/UMA/scripts/gif_maker_parallelized.py
src/onescience/models/UMA/scripts/make_challenge_submission_file.py
src/onescience/models/UMA/scripts/make_lmdb_sizes.py
src/onescience/models/UMA/scripts/make_submission_file.py
src/onescience/models/UMA/scripts/migrate_checkpoint.py
src/onescience/models/UMA/scripts/migrate_imports.py
src/onescience/models/UMA/scripts/preprocess_ef.py
src/onescience/models/UMA/scripts/preprocess_relaxed.py
src/onescience/models/UMA/scripts/uncompress.py
src/onescience/models/UMA/units/__init__.py
src/onescience/models/UMA/units/mlip_unit/__init__.py
src/onescience/models/UMA/units/mlip_unit/_metrics.py
src/onescience/models/UMA/units/mlip_unit/mlip_unit.py
src/onescience/models/UMA/units/mlip_unit/predict.py
src/onescience/models/UMA/units/mlip_unit/utils.py
src/onescience/models/UMA/units/mlip_unit/api/__init__.py
src/onescience/models/UMA/units/mlip_unit/api/inference.py
src/onescience/models/afno/__init__.py
src/onescience/models/afno/afnonet.py
src/onescience/models/afno/afnonet_dualpipe.py
src/onescience/models/afno/afnonet_ocean.py
src/onescience/models/afno/afnonet_pipeline.py
src/onescience/models/afno/distributed/__init__.py
src/onescience/models/afno/distributed/afno.py
src/onescience/models/afno/distributed/layers.py
src/onescience/models/beno/BE_MPNN.py
src/onescience/models/beno/__init__.py
src/onescience/models/beno/transformer.py
src/onescience/models/cfd_benchmark/DeepONet.py
src/onescience/models/cfd_benchmark/FNO.py
src/onescience/models/cfd_benchmark/F_FNO.py
src/onescience/models/cfd_benchmark/Factformer.py
src/onescience/models/cfd_benchmark/GFNO.py
src/onescience/models/cfd_benchmark/GNOT.py
src/onescience/models/cfd_benchmark/Galerkin_Transformer.py
src/onescience/models/cfd_benchmark/GraphSAGE.py
src/onescience/models/cfd_benchmark/Graph_UNet.py
src/onescience/models/cfd_benchmark/LSM.py
src/onescience/models/cfd_benchmark/MWT.py
src/onescience/models/cfd_benchmark/MeshGraphNet.py
src/onescience/models/cfd_benchmark/ONO.py
src/onescience/models/cfd_benchmark/PointNet.py
src/onescience/models/cfd_benchmark/RegDGCNN.py
src/onescience/models/cfd_benchmark/Swin_Transformer.py
src/onescience/models/cfd_benchmark/Transformer.py
src/onescience/models/cfd_benchmark/Transolver.py
src/onescience/models/cfd_benchmark/U_FNO.py
src/onescience/models/cfd_benchmark/U_NO.py
src/onescience/models/cfd_benchmark/U_Net.py
src/onescience/models/cfd_benchmark/__init__.py
src/onescience/models/cfd_benchmark/model_factory.py
src/onescience/models/cfdbench/__init__.py
src/onescience/models/cfdbench/act_fn.py
src/onescience/models/cfdbench/auto_deeponet.py
src/onescience/models/cfdbench/auto_deeponet_cnn.py
src/onescience/models/cfdbench/auto_edeeponet.py
src/onescience/models/cfdbench/auto_ffn.py
src/onescience/models/cfdbench/base_model.py
src/onescience/models/cfdbench/deeponet.py
src/onescience/models/cfdbench/ffn.py
src/onescience/models/cfdbench/loss.py
src/onescience/models/cfdbench/resnet.py
src/onescience/models/cfdbench/unet.py
src/onescience/models/deepcfd/AutoEncoder.py
src/onescience/models/deepcfd/AutoEncoderEx.py
src/onescience/models/deepcfd/UNet.py
src/onescience/models/deepcfd/UNetEx.py
src/onescience/models/deepcfd/UNetExAvg.py
src/onescience/models/deepcfd/UNetExMod.py
src/onescience/models/deepcfd/__init__.py
src/onescience/models/diffusion/__init__.py
src/onescience/models/diffusion/dhariwal_unet.py
src/onescience/models/diffusion/layers.py
src/onescience/models/diffusion/preconditioning.py
src/onescience/models/diffusion/song_unet.py
src/onescience/models/diffusion/unet.py
src/onescience/models/diffusion/utils.py
src/onescience/models/evo2/__init__.py
src/onescience/models/evo2/api.py
src/onescience/models/evo2/lightning.py
src/onescience/models/evo2/data/__init__.py
src/onescience/models/evo2/data/fasta_dataset.py
src/onescience/models/evo2/data/tokenizer.py
src/onescience/models/evo2/data/transcript_extraction.py
src/onescience/models/evo2/models/__init__.py
src/onescience/models/evo2/models/mamba.py
src/onescience/models/evo2/utils/__init__.py
src/onescience/models/evo2/utils/callbacks.py
src/onescience/models/evo2/utils/config.py
src/onescience/models/evo2/utils/datamodule_utils.py
src/onescience/models/evo2/utils/logger_utils.py
src/onescience/models/evo2/utils/checkpoint/__init__.py
src/onescience/models/evo2/utils/checkpoint/convert_checkpoint_model_parallel_evo2.py
src/onescience/models/evo2/utils/checkpoint/convert_to_nemo.py
src/onescience/models/evo2/utils/checkpoint/convert_zero3_to_zero1.py
src/onescience/models/evo2/utils/checkpoint/params.py
src/onescience/models/evo2/utils/checkpoint/zero3_conversion_lib.py
src/onescience/models/evo2/utils/logging/__init__.py
src/onescience/models/evo2/utils/logging/callbacks.py
src/onescience/models/evo2/utils/loss/__init__.py
src/onescience/models/evo2/utils/loss/embedding_variance.py
src/onescience/models/fengwu/__init__.py
src/onescience/models/fengwu/fengwu.py
src/onescience/models/fuxi/__init__.py
src/onescience/models/fuxi/fuxi.py
src/onescience/models/fuxi/pad.py
src/onescience/models/gnn_layers/__init__.py
src/onescience/models/gnn_layers/bsms.py
src/onescience/models/gnn_layers/distributed_graph.py
src/onescience/models/gnn_layers/embedder.py
src/onescience/models/gnn_layers/graph.py
src/onescience/models/gnn_layers/mesh_edge_block.py
src/onescience/models/gnn_layers/mesh_graph_decoder.py
src/onescience/models/gnn_layers/mesh_graph_encoder.py
src/onescience/models/gnn_layers/mesh_graph_mlp.py
src/onescience/models/gnn_layers/mesh_node_block.py
src/onescience/models/gnn_layers/utils.py
src/onescience/models/graphcast/__init__.py
src/onescience/models/graphcast/graph_cast_net.py
src/onescience/models/graphcast/graph_cast_processor.py
src/onescience/models/graphvit/__init__.py
src/onescience/models/graphvit/base.py
src/onescience/models/graphvit/graphViT.py
src/onescience/models/layers/Basic.py
src/onescience/models/layers/Embedding.py
src/onescience/models/layers/FFNO_Layers.py
src/onescience/models/layers/FNO_Layers.py
src/onescience/models/layers/GeoFNO_Projection.py
src/onescience/models/layers/MWT_Layers.py
src/onescience/models/layers/Neural_Spectral_Block.py
src/onescience/models/layers/Physics_Attention.py
src/onescience/models/layers/UNet_Blocks.py
src/onescience/models/layers/__init__.py
src/onescience/models/layers/activations.py
src/onescience/models/layers/attention_layers.py
src/onescience/models/layers/conv_layers.py
src/onescience/models/layers/dgm_layers.py
src/onescience/models/layers/drop.py
src/onescience/models/layers/fft.py
src/onescience/models/layers/fourier_layers.py
src/onescience/models/layers/fully_connected_layers.py
src/onescience/models/layers/fused_silu.py
src/onescience/models/layers/interpolation.py
src/onescience/models/layers/mlp_layers.py
src/onescience/models/layers/resample_layers.py
src/onescience/models/layers/siren_layers.py
src/onescience/models/layers/spectral_layers.py
src/onescience/models/layers/transformer_decoder.py
src/onescience/models/layers/transformer_layers.py
src/onescience/models/layers/weight_fact.py
src/onescience/models/layers/weight_norm.py
src/onescience/models/mace/__init__.py
src/onescience/models/mace/__version__.py
src/onescience/models/mace/py.typed
src/onescience/models/mace/calculators/__init__.py
src/onescience/models/mace/calculators/foundations_models.py
src/onescience/models/mace/calculators/lammps_mace.py
src/onescience/models/mace/calculators/lammps_mliap_mace.py
src/onescience/models/mace/calculators/mace.py
src/onescience/models/mace/cli/__init__.py
src/onescience/models/mace/cli/active_learning_md.py
src/onescience/models/mace/cli/convert_cueq_e3nn.py
src/onescience/models/mace/cli/convert_device.py
src/onescience/models/mace/cli/convert_e3nn_cueq.py
src/onescience/models/mace/cli/create_lammps_model.py
src/onescience/models/mace/cli/eval_configs.py
src/onescience/models/mace/cli/fine_tuning_select.py
src/onescience/models/mace/cli/plot_train.py
src/onescience/models/mace/cli/preprocess_data.py
src/onescience/models/mace/cli/run_train.py
src/onescience/models/mace/cli/select_head.py
src/onescience/models/mace/cli/visualise_train.py
src/onescience/models/mace/data/__init__.py
src/onescience/models/mace/data/atomic_data.py
src/onescience/models/mace/data/hdf5_dataset.py
src/onescience/models/mace/data/lmdb_dataset.py
src/onescience/models/mace/data/neighborhood.py
src/onescience/models/mace/data/utils.py
src/onescience/models/mace/modules/__init__.py
src/onescience/models/mace/modules/blocks.py
src/onescience/models/mace/modules/irreps_tools.py
src/onescience/models/mace/modules/loss.py
src/onescience/models/mace/modules/models.py
src/onescience/models/mace/modules/radial.py
src/onescience/models/mace/modules/symmetric_contraction.py
src/onescience/models/mace/modules/utils.py
src/onescience/models/mace/modules/wrapper_ops.py
src/onescience/models/mace/tools/__init__.py
src/onescience/models/mace/tools/arg_parser.py
src/onescience/models/mace/tools/arg_parser_tools.py
src/onescience/models/mace/tools/cg.py
src/onescience/models/mace/tools/checkpoint.py
src/onescience/models/mace/tools/compile.py
src/onescience/models/mace/tools/default_keys.py
src/onescience/models/mace/tools/finetuning_utils.py
src/onescience/models/mace/tools/model_script_utils.py
src/onescience/models/mace/tools/multihead_tools.py
src/onescience/models/mace/tools/run_train_utils.py
src/onescience/models/mace/tools/scatter.py
src/onescience/models/mace/tools/scripts_utils.py
src/onescience/models/mace/tools/slurm_distributed.py
src/onescience/models/mace/tools/tables_utils.py
src/onescience/models/mace/tools/torch_tools.py
src/onescience/models/mace/tools/train.py
src/onescience/models/mace/tools/utils.py
src/onescience/models/mace/tools/fairchem_dataset/__init__.py
src/onescience/models/mace/tools/fairchem_dataset/lmdb_dataset_tools.py
src/onescience/models/mace/tools/torch_geometric/__init__.py
src/onescience/models/mace/tools/torch_geometric/batch.py
src/onescience/models/mace/tools/torch_geometric/data.py
src/onescience/models/mace/tools/torch_geometric/dataloader.py
src/onescience/models/mace/tools/torch_geometric/dataset.py
src/onescience/models/mace/tools/torch_geometric/seed.py
src/onescience/models/mace/tools/torch_geometric/utils.py
src/onescience/models/meshgraphnet/__init__.py
src/onescience/models/meshgraphnet/bsms_mgn.py
src/onescience/models/meshgraphnet/meshgraphnet.py
src/onescience/models/mlp/__init__.py
src/onescience/models/mlp/fully_connected.py
src/onescience/models/nowcastnet/__init__.py
src/onescience/models/nowcastnet/discriminator2.py
src/onescience/models/nowcastnet/evolutionnet.py
src/onescience/models/nowcastnet/generator.py
src/onescience/models/nowcastnet/nowcastnet.py
src/onescience/models/openfold/__init__.py
src/onescience/models/openfold/dropout.py
src/onescience/models/openfold/outer_product_mean.py
src/onescience/models/openfold/primitives.py
src/onescience/models/openfold/triangular_attention.py
src/onescience/models/openfold/triangular_multiplicative_update.py
src/onescience/models/pangu/__init__.py
src/onescience/models/pangu/pangu.py
src/onescience/models/pdenneval/__init__.py
src/onescience/models/pdenneval/deeponet.py
src/onescience/models/pdenneval/fno.py
src/onescience/models/pdenneval/mpnn.py
src/onescience/models/pdenneval/pino_fno.py
src/onescience/models/pdenneval/unet.py
src/onescience/models/pdenneval/uno.py
src/onescience/models/pinnsformer/__init__.py
src/onescience/models/pinnsformer/fls.py
src/onescience/models/pinnsformer/pinn.py
src/onescience/models/pinnsformer/pinnsformer.py
src/onescience/models/pinnsformer/qres.py
src/onescience/models/pix2pix/__init__.py
src/onescience/models/pix2pix/pix2pix.py
src/onescience/models/proteinmpnn/__init__.py
src/onescience/models/proteinmpnn/model_utils.py
src/onescience/models/proteinmpnn/protein_mpnn_utils.py
src/onescience/models/proteinmpnn/utils.py
src/onescience/models/protenix/__init__.py
src/onescience/models/protenix/generator.py
src/onescience/models/protenix/loss.py
src/onescience/models/protenix/protenix.py
src/onescience/models/protenix/sample_confidence.py
src/onescience/models/protenix/utils.py
src/onescience/models/protenix/config/__init__.py
src/onescience/models/protenix/config/config.py
src/onescience/models/protenix/config/extend_types.py
src/onescience/models/protenix/layer_norm/__init__.py
src/onescience/models/protenix/layer_norm/layer_norm.py
src/onescience/models/protenix/layer_norm/torch_ext_compile.py
src/onescience/models/protenix/modules/__init__.py
src/onescience/models/protenix/modules/confidence.py
src/onescience/models/protenix/modules/diffusion.py
src/onescience/models/protenix/modules/embedders.py
src/onescience/models/protenix/modules/frames.py
src/onescience/models/protenix/modules/head.py
src/onescience/models/protenix/modules/pairformer.py
src/onescience/models/protenix/modules/primitives.py
src/onescience/models/protenix/modules/transformer.py
src/onescience/models/rfdiffusion/Attention_module.py
src/onescience/models/rfdiffusion/AuxiliaryPredictor.py
src/onescience/models/rfdiffusion/Embeddings.py
src/onescience/models/rfdiffusion/RoseTTAFoldModel.py
src/onescience/models/rfdiffusion/SE3_network.py
src/onescience/models/rfdiffusion/Track_module.py
src/onescience/models/rfdiffusion/__init__.py
src/onescience/models/rfdiffusion/diffusion.py
src/onescience/models/rfdiffusion/model_input_logger.py
src/onescience/models/rnn/__init__.py
src/onescience/models/rnn/layers.py
src/onescience/models/rnn/rnn_one2many.py
src/onescience/models/rnn/rnn_seq2seq.py
src/onescience/models/se3_transformer/__init__.py
src/onescience/models/se3_transformer/basis.py
src/onescience/models/se3_transformer/fiber.py
src/onescience/models/se3_transformer/transformer.py
src/onescience/models/se3_transformer/layers/__init__.py
src/onescience/models/se3_transformer/layers/attention.py
src/onescience/models/se3_transformer/layers/convolution.py
src/onescience/models/se3_transformer/layers/linear.py
src/onescience/models/se3_transformer/layers/norm.py
src/onescience/models/se3_transformer/layers/pooling.py
src/onescience/models/srrn/__init__.py
src/onescience/models/srrn/super_res_net.py
src/onescience/models/swinvrnn/__init__.py
src/onescience/models/swinvrnn/swinvrnn.py
src/onescience/models/transolver/GUNet.py
src/onescience/models/transolver/GraphSAGE.py
src/onescience/models/transolver/MLP.py
src/onescience/models/transolver/NN.py
src/onescience/models/transolver/PointNet.py
src/onescience/models/transolver/Transolver2D.py
src/onescience/models/transolver/Transolver3D.py
src/onescience/models/transolver/__init__.py
src/onescience/models/unet/__init__.py
src/onescience/models/unet/unet.py
src/onescience/models/utils/__init__.py
src/onescience/models/utils/patch_embed.py
src/onescience/models/utils/shift_window_mask.py
src/onescience/models/utils/utils.py
src/onescience/models/utils/weight_init.py
src/onescience/models/vfgn/__init__.py
src/onescience/models/vfgn/graph_network_modules.py
src/onescience/registry/__init__.py
src/onescience/registry/model_registry.py
src/onescience/sciui/__init__.py
src/onescience/sciui/alphafold/__init__.py
src/onescience/sciui/alphafold/web_service/__init__.py
src/onescience/sciui/alphafold/web_service/colab_request_parser.py
src/onescience/sciui/alphafold/web_service/colab_request_utils.py
src/onescience/sciui/alphafold/web_service/dependency_url.py
src/onescience/sciui/alphafold/web_service/prediction_visualization.py
src/onescience/sciui/alphafold/web_service/viewer.py
src/onescience/sciui/protenix/__init__.py
src/onescience/sciui/protenix/web_service/__init__.py
src/onescience/sciui/protenix/web_service/colab_request_parser.py
src/onescience/sciui/protenix/web_service/colab_request_utils.py
src/onescience/sciui/protenix/web_service/dependency_url.py
src/onescience/sciui/protenix/web_service/prediction_visualization.py
src/onescience/sciui/protenix/web_service/viewer.py
src/onescience/utils/__init__.py
src/onescience/utils/capture.py
src/onescience/utils/filesystem.py
src/onescience/utils/insolation.py
src/onescience/utils/neighbor_list.py
src/onescience/utils/pinnsformer_util.py
src/onescience/utils/sdf.py
src/onescience/utils/zenith_angle.py
src/onescience/utils/GP_TO/__init__.py
src/onescience/utils/GP_TO/gen_data.py
src/onescience/utils/GP_TO/input_space.py
src/onescience/utils/GP_TO/interval_score.py
src/onescience/utils/GP_TO/plot_latent.py
src/onescience/utils/GP_TO/plot_latenth.py
src/onescience/utils/GP_TO/plot_latenth_position.py
src/onescience/utils/GP_TO/set_seed.py
src/onescience/utils/GP_TO/transforms.py
src/onescience/utils/GP_TO/variables.py
src/onescience/utils/GP_TO/visualize.py
src/onescience/utils/beno/__init__.py
src/onescience/utils/beno/util.py
src/onescience/utils/beno/utilities.py
src/onescience/utils/cfd_benchmark/__init__.py
src/onescience/utils/cfd_benchmark/drag_coefficient.py
src/onescience/utils/cfd_benchmark/loss.py
src/onescience/utils/cfd_benchmark/normalizer.py
src/onescience/utils/cfd_benchmark/visual.py
src/onescience/utils/cfdbench/__init__.py
src/onescience/utils/cfdbench/get_result.py
src/onescience/utils/cfdbench/plot_multistep_inference.py
src/onescience/utils/cfdbench/test_multistep.py
src/onescience/utils/cfdbench/utils.py
src/onescience/utils/cfdbench/utils_auto.py
src/onescience/utils/cfdbench/dataset/__init__.py
src/onescience/utils/cfdbench/dataset/base.py
src/onescience/utils/cfdbench/dataset/cavity.py
src/onescience/utils/cfdbench/dataset/cylinder.py
src/onescience/utils/cfdbench/dataset/dam.py
src/onescience/utils/cfdbench/dataset/tube.py
src/onescience/utils/cfdbench/dataset/utils.py
src/onescience/utils/corrdiff/__init__.py
src/onescience/utils/corrdiff/utils.py
src/onescience/utils/deepcfd/__init__.py
src/onescience/utils/deepcfd/functions.py
src/onescience/utils/deepcfd/pytorchtools.py
src/onescience/utils/fcn/YParams.py
src/onescience/utils/fcn/__init__.py
src/onescience/utils/fcn/darcy_loss.py
src/onescience/utils/fcn/darcy_loss_ocean.py
src/onescience/utils/fcn/data_loader_multifiles.py
src/onescience/utils/fcn/data_loader_ocean.py
src/onescience/utils/fcn/date_time_to_hours.py
src/onescience/utils/fcn/img_utils.py
src/onescience/utils/fcn/logging_utils.py
src/onescience/utils/fcn/weighted_acc_rmse.py
src/onescience/utils/generative/__init__.py
src/onescience/utils/generative/deterministic_sampler.py
src/onescience/utils/generative/stochastic_sampler.py
src/onescience/utils/generative/utils.py
src/onescience/utils/graphcast/__init__.py
src/onescience/utils/graphcast/data_utils.py
src/onescience/utils/graphcast/graph.py
src/onescience/utils/graphcast/graph_utils.py
src/onescience/utils/graphcast/icosahedral_mesh.py
src/onescience/utils/graphcast/loss.py
src/onescience/utils/mesh/__init__.py
src/onescience/utils/mesh/combine_vtp_files.py
src/onescience/utils/mesh/convert_file_formats.py
src/onescience/utils/mesh/generate_stl.py
src/onescience/utils/openfold/__init__.py
src/onescience/utils/openfold/all_atom_multimer.py
src/onescience/utils/openfold/checkpointing.py
src/onescience/utils/openfold/chunk_utils.py
src/onescience/utils/openfold/feats.py
src/onescience/utils/openfold/precision_utils.py
src/onescience/utils/openfold/rigid_utils.py
src/onescience/utils/openfold/tensor_utils.py
src/onescience/utils/openfold/geometry/__init__.py
src/onescience/utils/openfold/geometry/quat_rigid.py
src/onescience/utils/openfold/geometry/rigid_matrix_vector.py
src/onescience/utils/openfold/geometry/rotation_matrix.py
src/onescience/utils/openfold/geometry/test_utils.py
src/onescience/utils/openfold/geometry/utils.py
src/onescience/utils/openfold/geometry/vector.py
src/onescience/utils/openfold/kernel/__init__.py
src/onescience/utils/openfold/kernel/attention_core.py
src/onescience/utils/openfold/np/__init__.py
src/onescience/utils/openfold/np/residue_constants.py
src/onescience/utils/pdenneval/__init__.py
src/onescience/utils/pdenneval/deeponet_utils.py
src/onescience/utils/pdenneval/dfvm_GenerateData.py
src/onescience/utils/pdenneval/fno_utils.py
src/onescience/utils/pdenneval/integral_operators.py
src/onescience/utils/pdenneval/metrics.py
src/onescience/utils/pdenneval/mpnn_utils.py
src/onescience/utils/pdenneval/pinn_metrics.py
src/onescience/utils/pdenneval/pinn_pde_definitions.py
src/onescience/utils/pdenneval/pinn_utils.py
src/onescience/utils/pdenneval/pino_loss.py
src/onescience/utils/pdenneval/pino_utils.py
src/onescience/utils/pdenneval/unet_utils.py
src/onescience/utils/pdenneval/uno_utils.py
src/onescience/utils/protenix/__init__.py
src/onescience/utils/protenix/cropping.py
src/onescience/utils/protenix/distributed.py
src/onescience/utils/protenix/file_io.py
src/onescience/utils/protenix/geometry.py
src/onescience/utils/protenix/logger.py
src/onescience/utils/protenix/lr_scheduler.py
src/onescience/utils/protenix/metrics.py
src/onescience/utils/protenix/scatter_utils.py
src/onescience/utils/protenix/seed.py
src/onescience/utils/protenix/torch_utils.py
src/onescience/utils/protenix/training.py
src/onescience/utils/protenix/permutation/__init__.py
src/onescience/utils/protenix/permutation/atom_permutation.py
src/onescience/utils/protenix/permutation/permutation.py
src/onescience/utils/protenix/permutation/utils.py
src/onescience/utils/protenix/permutation/chain_permutation/__init__.py
src/onescience/utils/protenix/permutation/chain_permutation/heuristic.py
src/onescience/utils/protenix/permutation/chain_permutation/pocket_based_permutation.py
src/onescience/utils/protenix/permutation/chain_permutation/utils.py
src/onescience/utils/rfdiffusion/__init__.py
src/onescience/utils/rfdiffusion/chemical.py
src/onescience/utils/rfdiffusion/contigs.py
src/onescience/utils/rfdiffusion/coords6d.py
src/onescience/utils/rfdiffusion/igso3.py
src/onescience/utils/rfdiffusion/kinematics.py
src/onescience/utils/rfdiffusion/scoring.py
src/onescience/utils/rfdiffusion/util.py
src/onescience/utils/rfdiffusion/util_module.py
src/onescience/utils/rfdiffusion/inference/__init__.py
src/onescience/utils/rfdiffusion/inference/model_runners.py
src/onescience/utils/rfdiffusion/inference/symmetry.py
src/onescience/utils/rfdiffusion/inference/utils.py
src/onescience/utils/rfdiffusion/potentials/__init__.py
src/onescience/utils/rfdiffusion/potentials/manager.py
src/onescience/utils/rfdiffusion/potentials/potentials.py
src/onescience/utils/se3_transformer/__init__.py
src/onescience/utils/se3_transformer/runtime/__init__.py
src/onescience/utils/se3_transformer/runtime/arguments.py
src/onescience/utils/se3_transformer/runtime/callbacks.py
src/onescience/utils/se3_transformer/runtime/gpu_affinity.py
src/onescience/utils/se3_transformer/runtime/inference.py
src/onescience/utils/se3_transformer/runtime/loggers.py
src/onescience/utils/se3_transformer/runtime/metrics.py
src/onescience/utils/se3_transformer/runtime/training.py
src/onescience/utils/se3_transformer/runtime/utils.py
src/onescience/utils/transolver/__init__.py
src/onescience/utils/transolver/drag_coefficient.py
src/onescience/utils/transolver/metrics.py
src/onescience/utils/transolver/metrics_NACA.py
src/onescience/utils/transolver/naca_generator.py
src/onescience/utils/transolver/reorganize.py
tests/test_trainer_dcu.py