Primary symbols
oryxoryx.coreoryx.core.FlatPrimitiveoryx.core.HigherOrderPrimitiveoryx.core.NonInvertibleErrororyx.core.Pytreeoryx.core.call_bindoryx.core.custom_inverseoryx.core.harvestoryx.core.ildjoryx.core.interpretersoryx.core.interpreters.harvestoryx.core.interpreters.harvest.HarvestTraceoryx.core.interpreters.harvest.HarvestTraceroryx.core.interpreters.harvest.harvestoryx.core.interpreters.harvest.nestoryx.core.interpreters.harvest.plantoryx.core.interpreters.harvest.reaporyx.core.interpreters.harvest.soworyx.core.interpreters.inverseoryx.core.interpreters.inverse.coreoryx.core.interpreters.inverse.core.InverseAndILDJoryx.core.interpreters.inverse.core.inverseoryx.core.interpreters.inverse.core.inverse_and_ildjoryx.core.interpreters.inverse.core.register_binaryoryx.core.interpreters.inverse.core.register_elementwiseoryx.core.interpreters.inverse.custom_inverseoryx.core.interpreters.inverse.custom_inverse.NonInvertibleErrororyx.core.interpreters.inverse.custom_inverse.custom_inverseoryx.core.interpreters.inverse.rulesoryx.core.interpreters.inverse.sliceoryx.core.interpreters.log_proboryx.core.interpreters.log_prob.LogProbRulesoryx.core.interpreters.log_prob.log_proboryx.core.interpreters.propagateoryx.core.interpreters.propagate.Celloryx.core.interpreters.propagate.Environmentoryx.core.interpreters.propagate.Equationoryx.core.interpreters.propagate.propagateoryx.core.interpreters.unziporyx.core.interpreters.unzip.UnzipTraceoryx.core.interpreters.unzip.UnzipTraceroryx.core.interpreters.unzip.VariableErrororyx.core.interpreters.unzip.unziporyx.core.inverseoryx.core.inverse_and_ildjoryx.core.log_proboryx.core.nestoryx.core.plantoryx.core.pploryx.core.ppl.LogProbFunctionoryx.core.ppl.Programoryx.core.ppl.conditionaloryx.core.ppl.graph_replaceoryx.core.ppl.interveneoryx.core.ppl.joint_log_proboryx.core.ppl.joint_sampleoryx.core.ppl.log_proboryx.core.ppl.random_variableoryx.core.ppl.transformationsoryx.core.ppl.transformations.conditionaloryx.core.ppl.transformations.graph_replaceoryx.core.ppl.transformations.interveneoryx.core.ppl.transformations.joint_log_proboryx.core.ppl.transformations.joint_sampleoryx.core.ppl.transformations.log_proboryx.core.ppl.transformations.random_variableoryx.core.primitiveoryx.core.primitive.FlatPrimitiveoryx.core.primitive.HigherOrderPrimitiveoryx.core.primitive.call_bindoryx.core.primitive.tie_alloryx.core.primitive.tie_inoryx.core.pytreeoryx.core.pytree.Pytreeoryx.core.reaporyx.core.serializeoryx.core.serialize.deserializeoryx.core.serialize.from_tupleoryx.core.serialize.serializeoryx.core.serialize.to_tupleoryx.core.soworyx.core.stateoryx.core.state.ArraySpecoryx.core.state.Moduleoryx.core.state.Shapeoryx.core.state.apioryx.core.state.api.ArraySpecoryx.core.state.api.calloryx.core.state.api.call_and_updateoryx.core.state.api.initoryx.core.state.api.make_array_specoryx.core.state.api.specoryx.core.state.api.updateoryx.core.state.assignoryx.core.state.calloryx.core.state.call_and_updateoryx.core.state.functionoryx.core.state.function.FunctionModuleoryx.core.state.initoryx.core.state.make_array_specoryx.core.state.moduleoryx.core.state.module.Moduleoryx.core.state.module.assignoryx.core.state.module.variableoryx.core.state.specoryx.core.state.updateoryx.core.state.variableoryx.core.tie_alloryx.core.tie_inoryx.core.trace_utiloryx.core.trace_util.get_dynamic_contextoryx.core.trace_util.get_shaped_avaloryx.core.trace_util.new_dynamic_contextoryx.core.trace_util.pv_likeoryx.core.trace_util.stageoryx.core.trace_util.treesoryx.core.unziporyx.experimentaloryx.experimental.mcmcoryx.experimental.mcmc.constrainoryx.experimental.mcmc.hmcoryx.experimental.mcmc.kernelsoryx.experimental.mcmc.kernels.hmcoryx.experimental.mcmc.kernels.malaoryx.experimental.mcmc.kernels.metropolisoryx.experimental.mcmc.kernels.metropolis_hastingsoryx.experimental.mcmc.kernels.random_walkoryx.experimental.mcmc.kernels.sample_chainoryx.experimental.mcmc.malaoryx.experimental.mcmc.metropolisoryx.experimental.mcmc.metropolis_hastingsoryx.experimental.mcmc.random_walkoryx.experimental.mcmc.sample_chainoryx.experimental.mcmc.utilsoryx.experimental.mcmc.utils.constrainoryx.experimental.nnoryx.experimental.nn.AvgPoolingoryx.experimental.nn.BatchNormoryx.experimental.nn.Convoryx.experimental.nn.Deconvoryx.experimental.nn.Denseoryx.experimental.nn.Dropoutoryx.experimental.nn.Flattenoryx.experimental.nn.Layeroryx.experimental.nn.LayerParamsoryx.experimental.nn.LogSoftmaxoryx.experimental.nn.MaxPoolingoryx.experimental.nn.Reluoryx.experimental.nn.Reshapeoryx.experimental.nn.Serialoryx.experimental.nn.Softmaxoryx.experimental.nn.Softplusoryx.experimental.nn.SumPoolingoryx.experimental.nn.Tanhoryx.experimental.nn.Templateoryx.experimental.nn.baseoryx.experimental.nn.base.Layeroryx.experimental.nn.base.LayerParamsoryx.experimental.nn.base.Templateoryx.experimental.nn.base.create_parameteroryx.experimental.nn.combinatororyx.experimental.nn.combinator.Serialoryx.experimental.nn.convolutionoryx.experimental.nn.convolution.Convoryx.experimental.nn.convolution.Deconvoryx.experimental.nn.convolution.conv_infooryx.experimental.nn.coreoryx.experimental.nn.core.Activationoryx.experimental.nn.core.Denseoryx.experimental.nn.core.Dropoutoryx.experimental.nn.core.LogSoftmaxoryx.experimental.nn.core.Reluoryx.experimental.nn.core.Softmaxoryx.experimental.nn.core.Softplusoryx.experimental.nn.core.Tanhoryx.experimental.nn.normalizationoryx.experimental.nn.normalization.BatchNormoryx.experimental.nn.poolingoryx.experimental.nn.pooling.AvgPoolingoryx.experimental.nn.pooling.MaxPoolingoryx.experimental.nn.pooling.Poolingoryx.experimental.nn.pooling.SumPoolingoryx.experimental.nn.reshapeoryx.experimental.nn.reshape.Flattenoryx.experimental.nn.reshape.Reshapeoryx.experimental.optimizersoryx.experimental.optimizers.adamoryx.experimental.optimizers.gradient_descentoryx.experimental.optimizers.noisy_sgdoryx.experimental.optimizers.optimizeoryx.experimental.optimizers.optixoryx.experimental.optimizers.optix.adamoryx.experimental.optimizers.optix.add_noiseoryx.experimental.optimizers.optix.apply_everyoryx.experimental.optimizers.optix.chainoryx.experimental.optimizers.optix.cliporyx.experimental.optimizers.optix.clip_by_global_normoryx.experimental.optimizers.optix.global_normoryx.experimental.optimizers.optix.gradient_descentoryx.experimental.optimizers.optix.noisy_sgdoryx.experimental.optimizers.optix.optimizeoryx.experimental.optimizers.optix.rmsproporyx.experimental.optimizers.optix.scaleoryx.experimental.optimizers.optix.scale_by_adamoryx.experimental.optimizers.optix.scale_by_rmsoryx.experimental.optimizers.optix.scale_by_scheduleoryx.experimental.optimizers.optix.scale_by_stddevoryx.experimental.optimizers.optix.sgdoryx.experimental.optimizers.optix.traceoryx.experimental.optimizers.rmsproporyx.experimental.optimizers.sgdoryx.version