LICENSE
README.md
pyproject.toml
src/pyjuice/__init__.py
src/pyjuice.egg-info/PKG-INFO
src/pyjuice.egg-info/SOURCES.txt
src/pyjuice.egg-info/dependency_links.txt
src/pyjuice.egg-info/requires.txt
src/pyjuice.egg-info/top_level.txt
src/pyjuice/distributions/__init__.py
src/pyjuice/graph/__init__.py
src/pyjuice/graph/region_graph.py
src/pyjuice/io/__init__.py
src/pyjuice/io/io.py
src/pyjuice/io/serialization.py
src/pyjuice/layer/__init__.py
src/pyjuice/layer/compilation.py
src/pyjuice/layer/external_sum_layer.py
src/pyjuice/layer/input_layer.py
src/pyjuice/layer/layer.py
src/pyjuice/layer/layer_group.py
src/pyjuice/layer/prod_layer.py
src/pyjuice/layer/sum_layer.py
src/pyjuice/layer/backend/__init__.py
src/pyjuice/layer/backend/index_set.py
src/pyjuice/layer/backend/node_partition.py
src/pyjuice/layer/kernels/__init__.py
src/pyjuice/layer/kernels/prod.py
src/pyjuice/layer/kernels/sum_backward_element_block_sparse.py
src/pyjuice/layer/kernels/sum_backward_element_sparse.py
src/pyjuice/layer/kernels/sum_backward_node_flows.py
src/pyjuice/layer/kernels/sum_backward_param_block_sparse.py
src/pyjuice/layer/kernels/sum_backward_param_sparse.py
src/pyjuice/layer/kernels/sum_forward_block_sparse.py
src/pyjuice/layer/kernels/sum_forward_sparse.py
src/pyjuice/layer/kernels/c/__init__.py
src/pyjuice/layer/kernels/c/ele_backward_sum.cu
src/pyjuice/layer/kernels/c/par_backward_sum.cu
src/pyjuice/layer/kernels/c/smallbatch_ele_backward.cu
src/pyjuice/layer/kernels/c/smallbatch_forward_sum.cu
src/pyjuice/layer/kernels/c/smallbatch_par_backward.cu
src/pyjuice/layer/kernels/c/tlmm_forward_sum.cu
src/pyjuice/layer/kernels/c/input_layers/__init__.py
src/pyjuice/model/__init__.py
src/pyjuice/model/tensorcircuit.py
src/pyjuice/model/backend/__init__.py
src/pyjuice/model/backend/eval_partition.py
src/pyjuice/model/backend/normalize.py
src/pyjuice/model/backend/par_update.py
src/pyjuice/model/backend/parflow_fusing.py
src/pyjuice/model/backend/partition_grad.py
src/pyjuice/model/backend/top_down_prob.py
src/pyjuice/nodes/__init__.py
src/pyjuice/nodes/construction.py
src/pyjuice/nodes/external_sum_nodes.py
src/pyjuice/nodes/input_nodes.py
src/pyjuice/nodes/nodes.py
src/pyjuice/nodes/prod_nodes.py
src/pyjuice/nodes/sum_nodes.py
src/pyjuice/nodes/backend/__init__.py
src/pyjuice/nodes/backend/normalize.py
src/pyjuice/nodes/distributions/__init__.py
src/pyjuice/nodes/distributions/bernoulli.py
src/pyjuice/nodes/distributions/categorical.py
src/pyjuice/nodes/distributions/discrete_logistic.py
src/pyjuice/nodes/distributions/distributions.py
src/pyjuice/nodes/distributions/external.py
src/pyjuice/nodes/distributions/external_categorical.py
src/pyjuice/nodes/distributions/gaussian.py
src/pyjuice/nodes/distributions/indicator.py
src/pyjuice/nodes/distributions/latent_softevi.py
src/pyjuice/nodes/distributions/literal.py
src/pyjuice/nodes/distributions/masked_categorical.py
src/pyjuice/nodes/distributions/softevi_categorical.py
src/pyjuice/nodes/distributions/softevi_indicator.py
src/pyjuice/nodes/distributions/c_kernels/__init__.py
src/pyjuice/nodes/external_params/__init__.py
src/pyjuice/nodes/external_params/block_scale.py
src/pyjuice/nodes/external_params/external_params.py
src/pyjuice/nodes/external_params/lowrank.py
src/pyjuice/nodes/external_params/kernels/__init__.py
src/pyjuice/nodes/external_params/kernels/blockscale_backward.py
src/pyjuice/nodes/external_params/kernels/blockscale_forward.py
src/pyjuice/nodes/external_params/kernels/lowrank_forward.py
src/pyjuice/nodes/external_params/kernels/staging.py
src/pyjuice/nodes/external_params/kernels/c/__init__.py
src/pyjuice/nodes/methods/__init__.py
src/pyjuice/nodes/methods/edge_constructors.py
src/pyjuice/nodes/methods/lvd.py
src/pyjuice/nodes/methods/scope.py
src/pyjuice/nodes/methods/traversal.py
src/pyjuice/nodes/methods/lvd_backend/__init__.py
src/pyjuice/nodes/methods/lvd_backend/counting.py
src/pyjuice/optim/__init__.py
src/pyjuice/optim/adam.py
src/pyjuice/optim/anemone.py
src/pyjuice/optim/full_batch_em.py
src/pyjuice/optim/gradient.py
src/pyjuice/optim/mini_batch_em.py
src/pyjuice/optim/optim.py
src/pyjuice/optim/sgd.py
src/pyjuice/queries/__init__.py
src/pyjuice/queries/base.py
src/pyjuice/queries/conditional.py
src/pyjuice/queries/marginal.py
src/pyjuice/queries/sample.py
src/pyjuice/structures/__init__.py
src/pyjuice/structures/compilation.py
src/pyjuice/structures/hclt.py
src/pyjuice/structures/hmm.py
src/pyjuice/structures/pd.py
src/pyjuice/structures/rat_spn.py
src/pyjuice/transformations/__init__.py
src/pyjuice/transformations/blockify.py
src/pyjuice/transformations/copy.py
src/pyjuice/transformations/grow.py
src/pyjuice/transformations/merge.py
src/pyjuice/transformations/prune.py
src/pyjuice/utils/__init__.py
src/pyjuice/utils/bitset.py
src/pyjuice/utils/context_manager.py
src/pyjuice/utils/cuda_ext.py
src/pyjuice/utils/grad_fns.py
src/pyjuice/utils/kernel_launcher.py
src/pyjuice/utils/kmeans.py
src/pyjuice/utils/parameter_list.py
src/pyjuice/utils/source2fn.py
src/pyjuice/utils/util.py
src/pyjuice/visualize/__init__.py
src/pyjuice/visualize/visualize.py