gitextract_9k1q0bj_/

├── .github/
│   └── workflows/
│       ├── release.yml
│       ├── scripts/
│       │   ├── release_linux.sh
│       │   ├── release_osx.sh
│       │   └── release_windows.bat
│       └── test.yml
├── .gitignore
├── .pre-commit-config.yaml
├── LICENSE
├── README.md
├── catenets/
│   ├── __init__.py
│   ├── datasets/
│   │   ├── __init__.py
│   │   ├── dataset_acic2016.py
│   │   ├── dataset_ihdp.py
│   │   ├── dataset_twins.py
│   │   └── network.py
│   ├── experiment_utils/
│   │   ├── __init__.py
│   │   ├── base.py
│   │   ├── simulation_utils.py
│   │   ├── tester.py
│   │   └── torch_metrics.py
│   ├── logger.py
│   ├── models/
│   │   ├── __init__.py
│   │   ├── constants.py
│   │   ├── jax/
│   │   │   ├── __init__.py
│   │   │   ├── base.py
│   │   │   ├── disentangled_nets.py
│   │   │   ├── flextenet.py
│   │   │   ├── model_utils.py
│   │   │   ├── offsetnet.py
│   │   │   ├── pseudo_outcome_nets.py
│   │   │   ├── representation_nets.py
│   │   │   ├── rnet.py
│   │   │   ├── snet.py
│   │   │   ├── tnet.py
│   │   │   ├── transformation_utils.py
│   │   │   └── xnet.py
│   │   └── torch/
│   │       ├── __init__.py
│   │       ├── base.py
│   │       ├── flextenet.py
│   │       ├── pseudo_outcome_nets.py
│   │       ├── representation_nets.py
│   │       ├── slearner.py
│   │       ├── snet.py
│   │       ├── tlearner.py
│   │       └── utils/
│   │           ├── __init__.py
│   │           ├── decorators.py
│   │           ├── model_utils.py
│   │           ├── transformations.py
│   │           └── weight_utils.py
│   └── version.py
├── docs/
│   ├── Makefile
│   ├── conf.py
│   ├── datasets.rst
│   ├── index.rst
│   ├── jax_models.rst
│   ├── make.bat
│   ├── requirements.txt
│   └── torch_models.rst
├── experiments/
│   ├── __init__.py
│   ├── experiments_AISTATS21/
│   │   ├── ihdp_experiments.py
│   │   └── simulations_AISTATS.py
│   ├── experiments_benchmarks_NeurIPS21/
│   │   ├── README.md
│   │   ├── __init__.py
│   │   ├── acic_experiments_catenets.py
│   │   ├── acic_experiments_grf.R
│   │   ├── ihdp_experiments_catenets.py
│   │   ├── ihdp_experiments_grf.R
│   │   ├── twins_experiments_catenets.py
│   │   └── twins_experiments_grf.R
│   └── experiments_inductivebias_NeurIPS21/
│       ├── __init__.py
│       ├── experiments_AB.py
│       ├── experiments_CD.py
│       ├── experiments_acic.py
│       └── experiments_twins.py
├── pyproject.toml
├── pytest.ini
├── run_experiments_AISTATS.py
├── run_experiments_benchmarks_NeurIPS.py
├── run_experiments_inductive_bias_NeurIPS.py
├── setup.py
└── tests/
    ├── conftest.py
    ├── datasets/
    │   └── test_datasets.py
    └── models/
        ├── jax/
        │   ├── test_jax_ite.py
        │   ├── test_jax_model_utils.py
        │   └── test_jax_transformation_utils.py
        └── torch/
            ├── test_torch_flextenet.py
            ├── test_torch_pseudo_outcome_nets.py
            ├── test_torch_representation_net.py
            ├── test_torch_slearner.py
            ├── test_torch_snet.py
            └── test_torch_tlearner.py