LICENSE
README.md
pyproject.toml
tests/test_amortized_scale.py
tests/test_amortized_training.py
tests/test_asym_scatter.py
tests/test_bg_mean_offset.py
tests/test_cli_pipeline.py
tests/test_cli_production.py
tests/test_core.py
tests/test_duckdb_load.py
tests/test_ewald_offset.py
tests/test_filter.py
tests/test_geometry.py
tests/test_geometry_reciprocal.py
tests/test_grid_tooling.py
tests/test_hkl.py
tests/test_io.py
tests/test_large_batch_diagnostics.py
tests/test_levels.py
tests/test_loo_pooling.py
tests/test_marginal_normalisation.py
tests/test_measurement_sources.py
tests/test_merge_stats.py
tests/test_merging_layers.py
tests/test_metadata.py
tests/test_native_index.py
tests/test_native_parse.py
tests/test_observation_filter.py
tests/test_pad.py
tests/test_peer_cross_attention.py
tests/test_peer_features.py
tests/test_posterior_nodes.py
tests/test_posterior_prior.py
tests/test_posterior_smooth_sigma.py
tests/test_reciprocal_basis.py
tests/test_scatter_mixture.py
tests/test_shape_prior.py
tests/test_split_gaussian_mean.py
tests/test_split_half.py
tests/test_split_log_grid.py
tests/test_split_mixture.py
tests/test_stateless_merge.py
tests/test_stats.py
tests/test_stream.py
tests/test_train_schedule.py
tests/test_wilson_surface.py
tests/test_writeback.py
tests/test_writeback_native.py
torchsx/__init__.py
torchsx/_categorical.py
torchsx/_core.py
torchsx/geometry.py
torchsx/levels.py
torchsx/measurement.py
torchsx/merge_stats.py
torchsx/stats.py
torchsx/stream.py
torchsx.egg-info/PKG-INFO
torchsx.egg-info/SOURCES.txt
torchsx.egg-info/dependency_links.txt
torchsx.egg-info/entry_points.txt
torchsx.egg-info/requires.txt
torchsx.egg-info/top_level.txt
torchsx/cli/__init__.py
torchsx/cli/_pipeline.py
torchsx/cli/merge_nn.py
torchsx/cli/merge_stats.py
torchsx/cli/stats.py
torchsx/cli/train.py
torchsx/io/__init__.py
torchsx/io/_crystfel.py
torchsx/io/_duckdb.py
torchsx/io/_metadata.py
torchsx/io/_native.py
torchsx/io/_parser.py
torchsx/io/_safetensors.py
torchsx/io/_writeback.py
torchsx/io/sources.py
torchsx/io/times.py
torchsx/io/_parse/__init__.py
torchsx/io/_parse/indexing.py
torchsx/io/_parse/orchestrator.py
torchsx/io/_parse/records.py
torchsx/io/_parse/workers.py
torchsx/merging/__init__.py
torchsx/merging/config.py
torchsx/merging/observations.py
torchsx/merging/parameters.py
torchsx/merging/scatter_shape.py
torchsx/merging/hkl/__init__.py
torchsx/merging/hkl/adjacency.py
torchsx/merging/hkl/filter.py
torchsx/merging/hkl/hkl.py
torchsx/merging/hkl/noise.py
torchsx/merging/hkl/shell_fit.py
torchsx/merging/hkl/stats.py
torchsx/merging/likelihood/__init__.py
torchsx/merging/likelihood/base.py
torchsx/merging/likelihood/detector_noise.py
torchsx/merging/likelihood/poisson.py
torchsx/merging/priors/__init__.py
torchsx/merging/priors/reciprocal_basis.py
torchsx/merging/quadrature/__init__.py
torchsx/merging/quadrature/_fixed_log_grid_triton.py
torchsx/merging/quadrature/_fixed_log_grid_tuning.json
torchsx/merging/quadrature/_fused_posterior_triton.py
torchsx/merging/quadrature/_posterior_nodes_triton.py
torchsx/merging/quadrature/_split_log_grid_triton.py
torchsx/merging/quadrature/_split_mixture_triton.py
torchsx/merging/quadrature/base.py
torchsx/merging/quadrature/fixed_log_grid.py
torchsx/merging/quadrature/split_log_grid.py
torchsx/merging/scale/__init__.py
torchsx/merging/scale/amortized.py
torchsx/merging/scale/base.py
torchsx/merging/scatter/__init__.py
torchsx/merging/scatter/base.py
torchsx/merging/scatter/gengauss.py
torchsx/merging/scatter/mixture.py
torchsx/merging/scatter/predicted.py
torchsx/nn/__init__.py
torchsx/nn/deepsets.py
torchsx/nn/heads.py
torchsx/nn/merge_target.py
torchsx/nn/metrics.py
torchsx/nn/objective.py
torchsx/nn/obs_features.py
torchsx/nn/pad.py
torchsx/nn/pooling.py
torchsx/nn/posterior_target.py
torchsx/nn/shape_prior.py
torchsx/nn/train_amortized.py