├── .bazelrc ├── .gitignore ├── .pylintrc ├── .style.yapf ├── .vscode └── settings.json ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── WORKSPACE.bazel ├── codex ├── __init__.py ├── cc │ ├── BUILD.bazel │ ├── range_ans.cc │ ├── range_ans.h │ ├── range_ans_pybind.cc │ ├── range_ans_pybind.pyi │ ├── range_ans_pybind_test.py │ └── range_ans_test.cc ├── ecs │ ├── __init__.py │ ├── range_ans.py │ └── range_ans_test.py ├── ems │ ├── __init__.py │ ├── continuous.py │ ├── deep_factorized.py │ ├── deep_factorized_test.py │ ├── distribution.py │ ├── distribution_test.py │ ├── equinox.py │ ├── equinox_test.py │ ├── flax.py │ ├── flax_test.py │ ├── fourier.py │ └── fourier_test.py ├── loss │ ├── __init__.py │ └── wasserstein.py └── ops │ ├── __init__.py │ ├── activation.py │ ├── activation_test.py │ ├── gradient.py │ ├── gradient_test.py │ ├── quantization.py │ └── quantization_test.py ├── examples └── ntc │ ├── config.py │ ├── ntc.py │ ├── train.py │ ├── train_lib.py │ └── train_test.py ├── pyproject.toml ├── tools ├── build_pip_package.sh └── test_pip_package.sh └── tutorial └── Fourier_Basis_Density_Estimation_Demo.ipynb /.bazelrc: -------------------------------------------------------------------------------- 1 | build --cxxopt=-std=c++17 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/.pylintrc -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/.style.yapf -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/README.md -------------------------------------------------------------------------------- /WORKSPACE.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/WORKSPACE.bazel -------------------------------------------------------------------------------- /codex/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/__init__.py -------------------------------------------------------------------------------- /codex/cc/BUILD.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/BUILD.bazel -------------------------------------------------------------------------------- /codex/cc/range_ans.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans.cc -------------------------------------------------------------------------------- /codex/cc/range_ans.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans.h -------------------------------------------------------------------------------- /codex/cc/range_ans_pybind.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans_pybind.cc -------------------------------------------------------------------------------- /codex/cc/range_ans_pybind.pyi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans_pybind.pyi -------------------------------------------------------------------------------- /codex/cc/range_ans_pybind_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans_pybind_test.py -------------------------------------------------------------------------------- /codex/cc/range_ans_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/cc/range_ans_test.cc -------------------------------------------------------------------------------- /codex/ecs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ecs/__init__.py -------------------------------------------------------------------------------- /codex/ecs/range_ans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ecs/range_ans.py -------------------------------------------------------------------------------- /codex/ecs/range_ans_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ecs/range_ans_test.py -------------------------------------------------------------------------------- /codex/ems/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/__init__.py -------------------------------------------------------------------------------- /codex/ems/continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/continuous.py -------------------------------------------------------------------------------- /codex/ems/deep_factorized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/deep_factorized.py -------------------------------------------------------------------------------- /codex/ems/deep_factorized_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/deep_factorized_test.py -------------------------------------------------------------------------------- /codex/ems/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/distribution.py -------------------------------------------------------------------------------- /codex/ems/distribution_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/distribution_test.py -------------------------------------------------------------------------------- /codex/ems/equinox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/equinox.py -------------------------------------------------------------------------------- /codex/ems/equinox_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/equinox_test.py -------------------------------------------------------------------------------- /codex/ems/flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/flax.py -------------------------------------------------------------------------------- /codex/ems/flax_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/flax_test.py -------------------------------------------------------------------------------- /codex/ems/fourier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/fourier.py -------------------------------------------------------------------------------- /codex/ems/fourier_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ems/fourier_test.py -------------------------------------------------------------------------------- /codex/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/loss/__init__.py -------------------------------------------------------------------------------- /codex/loss/wasserstein.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/loss/wasserstein.py -------------------------------------------------------------------------------- /codex/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/__init__.py -------------------------------------------------------------------------------- /codex/ops/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/activation.py -------------------------------------------------------------------------------- /codex/ops/activation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/activation_test.py -------------------------------------------------------------------------------- /codex/ops/gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/gradient.py -------------------------------------------------------------------------------- /codex/ops/gradient_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/gradient_test.py -------------------------------------------------------------------------------- /codex/ops/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/quantization.py -------------------------------------------------------------------------------- /codex/ops/quantization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/codex/ops/quantization_test.py -------------------------------------------------------------------------------- /examples/ntc/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/examples/ntc/config.py -------------------------------------------------------------------------------- /examples/ntc/ntc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/examples/ntc/ntc.py -------------------------------------------------------------------------------- /examples/ntc/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/examples/ntc/train.py -------------------------------------------------------------------------------- /examples/ntc/train_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/examples/ntc/train_lib.py -------------------------------------------------------------------------------- /examples/ntc/train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/examples/ntc/train_test.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tools/build_pip_package.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/tools/build_pip_package.sh -------------------------------------------------------------------------------- /tools/test_pip_package.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/tools/test_pip_package.sh -------------------------------------------------------------------------------- /tutorial/Fourier_Basis_Density_Estimation_Demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google/codex/HEAD/tutorial/Fourier_Basis_Density_Estimation_Demo.ipynb --------------------------------------------------------------------------------