gradgen
- gradgen package
- Submodules
- gradgen.ad module
- gradgen.composed_function module
ComposedGradientFunctionComposedJacobianFunctionComposedJointFunctionComposedJointFunction.composedComposedJointFunction.nameComposedJointFunction.componentsComposedJointFunction.wrt_indexComposedJointFunction.simplificationComposedJointFunction.to_function()ComposedJointFunction.nodesComposedJointFunction.generate_rust()ComposedJointFunction.create_rust_project()
ComposedFunctionComposedFunction.nameComposedFunction.state_inputComposedFunction.input_nameComposedFunction.parameter_nameComposedFunction.simplificationComposedFunction.stepsComposedFunction.finishedComposedFunction.parameter_sizeComposedFunction.stage_countComposedFunction.inputsComposedFunction.outputsComposedFunction.nodesComposedFunction.input_namesComposedFunction.output_namesComposedFunction.then()ComposedFunction.chain()ComposedFunction.repeat()ComposedFunction.finish()ComposedFunction.to_function()ComposedFunction.gradient()ComposedFunction.jacobian()ComposedFunction.joint()ComposedFunction.generate_rust()ComposedFunction.create_rust_project()
- gradgen.composer module
FunctionComposerFunctionCompositionFunctionComposition.nameFunctionComposition.stagesFunctionComposition.input_namesFunctionComposition.output_namesFunctionComposition.nodesFunctionComposition.to_function()FunctionComposition.gradient()FunctionComposition.hessian()FunctionComposition.hvp()FunctionComposition.generate_rust()FunctionComposition.create_rust_project()
- gradgen.cse module
- gradgen.function module
FunctionFunction.nameFunction.inputsFunction.outputsFunction.input_namesFunction.output_namesFunction.flat_inputsFunction.flat_outputsFunction.nodesFunction.cse()Function.generate_rust()Function.create_rust_project()Function.create_rust_derivative_bundle()Function.jvp()Function.gradient()Function.joint()Function.hvp()Function.vjp()Function.jacobian()Function.jacobian_blocks()Function.vjp_blocks()Function.hessian()Function.hessian_blocks()Function.hvp_blocks()Function.simplify()
- gradgen.map_zip module
BatchedJacobianFunctionBatchedJacobianFunction.batchedBatchedJacobianFunction.wrt_indexBatchedJacobianFunction.nameBatchedJacobianFunction.simplificationBatchedJacobianFunction.batchedBatchedJacobianFunction.wrt_indexBatchedJacobianFunction.nameBatchedJacobianFunction.simplificationBatchedJacobianFunction.to_function()BatchedJacobianFunction.input_namesBatchedJacobianFunction.output_namesBatchedJacobianFunction.generate_rust()BatchedJacobianFunction.create_rust_project()
BatchedFunctionBatchedFunction.functionBatchedFunction.countBatchedFunction.nameBatchedFunction.input_sequence_namesBatchedFunction.simplificationBatchedFunction.functionBatchedFunction.countBatchedFunction.nameBatchedFunction.input_sequence_namesBatchedFunction.simplificationBatchedFunction.input_namesBatchedFunction.output_namesBatchedFunction.nodesBatchedFunction.jacobian()BatchedFunction.to_function()BatchedFunction.generate_rust()BatchedFunction.create_rust_project()
ReducedFunctionReducedFunction.functionReducedFunction.countReducedFunction.nameReducedFunction.accumulator_input_nameReducedFunction.input_nameReducedFunction.output_nameReducedFunction.simplificationReducedFunction.functionReducedFunction.countReducedFunction.nameReducedFunction.accumulator_input_nameReducedFunction.input_nameReducedFunction.output_nameReducedFunction.simplificationReducedFunction.input_namesReducedFunction.output_namesReducedFunction.nodesReducedFunction.to_function()ReducedFunction.generate_rust()ReducedFunction.create_rust_project()
map_function()zip_function()reduce_function()
- gradgen.simplify module
- gradgen.single_shooting module
SingleShootingBundleSingleShootingPrimalFunctionSingleShootingPrimalFunction.problemSingleShootingPrimalFunction.nameSingleShootingPrimalFunction.include_statesSingleShootingPrimalFunction.simplificationSingleShootingPrimalFunction.to_function()SingleShootingPrimalFunction.nodesSingleShootingPrimalFunction.input_namesSingleShootingPrimalFunction.output_namesSingleShootingPrimalFunction.generate_rust()SingleShootingPrimalFunction.create_rust_project()
SingleShootingGradientFunctionSingleShootingGradientFunction.problemSingleShootingGradientFunction.nameSingleShootingGradientFunction.include_statesSingleShootingGradientFunction.simplificationSingleShootingGradientFunction.to_function()SingleShootingGradientFunction.nodesSingleShootingGradientFunction.input_namesSingleShootingGradientFunction.output_namesSingleShootingGradientFunction.generate_rust()SingleShootingGradientFunction.create_rust_project()
SingleShootingHvpFunctionSingleShootingHvpFunction.problemSingleShootingHvpFunction.nameSingleShootingHvpFunction.include_statesSingleShootingHvpFunction.simplificationSingleShootingHvpFunction.to_function()SingleShootingHvpFunction.nodesSingleShootingHvpFunction.input_namesSingleShootingHvpFunction.output_namesSingleShootingHvpFunction.generate_rust()SingleShootingHvpFunction.create_rust_project()
SingleShootingJointFunctionSingleShootingJointFunction.problemSingleShootingJointFunction.bundleSingleShootingJointFunction.nameSingleShootingJointFunction.simplificationSingleShootingJointFunction.to_function()SingleShootingJointFunction.nodesSingleShootingJointFunction.input_namesSingleShootingJointFunction.output_namesSingleShootingJointFunction.generate_rust()SingleShootingJointFunction.create_rust_project()
SingleShootingProblemSingleShootingProblem.nameSingleShootingProblem.horizonSingleShootingProblem.dynamicsSingleShootingProblem.stage_costSingleShootingProblem.terminal_costSingleShootingProblem.initial_state_nameSingleShootingProblem.control_sequence_nameSingleShootingProblem.parameter_nameSingleShootingProblem.simplificationSingleShootingProblem.stage_penaltySingleShootingProblem.terminal_penaltySingleShootingProblem.penalty_weightSingleShootingProblem.with_horizon()SingleShootingProblem.with_dynamics()SingleShootingProblem.with_stage_cost()SingleShootingProblem.with_terminal_cost()SingleShootingProblem.with_costs()SingleShootingProblem.with_penalties()SingleShootingProblem.with_input_names()SingleShootingProblem.with_simplification()SingleShootingProblem.state_sizeSingleShootingProblem.control_sizeSingleShootingProblem.parameter_sizeSingleShootingProblem.has_runtime_penalty_weightSingleShootingProblem.penalty_weight_nameSingleShootingProblem.input_namesSingleShootingProblem.output_namesSingleShootingProblem.inputsSingleShootingProblem.outputsSingleShootingProblem.nodesSingleShootingProblem.primal()SingleShootingProblem.gradient()SingleShootingProblem.hvp()SingleShootingProblem.joint()SingleShootingProblem.to_function()SingleShootingProblem.stage_total_cost_function()SingleShootingProblem.terminal_total_cost_function()SingleShootingProblem.generate_rust()SingleShootingProblem.create_rust_project()
- gradgen.squared_distance_to_set module
SquaredDistanceToSetSquaredDistanceToSet.nameSquaredDistanceToSet.with_sq_distance()SquaredDistanceToSet.with_sq_distance_function()SquaredDistanceToSet.with_projection()SquaredDistanceToSet.with_projection_function()SquaredDistanceToSet.with_rust_sq_distance()SquaredDistanceToSet.with_rust_projection()SquaredDistanceToSet.euclidean_ball()SquaredDistanceToSet.infinity_ball()SquaredDistanceToSet.rectangle()SquaredDistanceToSet.second_order_cone()SquaredDistanceToSet.to_function()SquaredDistanceToSet.gradient()SquaredDistanceToSet.jacobian()
- gradgen.sx module
SXNodeSXSX.nodeSX.sym()SX.const()SX.opSX.argsSX.nameSX.valueSX.metadataSX.sin()SX.cos()SX.tan()SX.asin()SX.acos()SX.atan()SX.atan2()SX.sinh()SX.cosh()SX.tanh()SX.asinh()SX.acosh()SX.atanh()SX.exp()SX.expm1()SX.log()SX.log1p()SX.sqrt()SX.cbrt()SX.erf()SX.erfc()SX.floor()SX.ceil()SX.round()SX.trunc()SX.fract()SX.signum()SX.hypot()SX.abs()SX.maximum()SX.minimum()
SXVectorSXVector.elementsSXVector.sym()SXVector.sin()SXVector.cos()SXVector.tan()SXVector.asin()SXVector.acos()SXVector.atan()SXVector.asinh()SXVector.acosh()SXVector.atanh()SXVector.sinh()SXVector.cosh()SXVector.tanh()SXVector.exp()SXVector.expm1()SXVector.log()SXVector.log1p()SXVector.sqrt()SXVector.cbrt()SXVector.erf()SXVector.erfc()SXVector.floor()SXVector.ceil()SXVector.round()SXVector.trunc()SXVector.fract()SXVector.signum()SXVector.abs()SXVector.dot()SXVector.cross()SXVector.sum()SXVector.prod()SXVector.max()SXVector.min()SXVector.mean()SXVector.norm2()SXVector.norm2sq()SXVector.norm1()SXVector.norm_inf()SXVector.norm_p()SXVector.norm_p_to_p()
const()sin()cos()tan()asin()acos()atan()atan2()sinh()cosh()tanh()asinh()acosh()atanh()exp()expm1()log()log1p()sqrt()cbrt()erf()erfc()floor()ceil()round()trunc()fract()signum()hypot()maximum()minimum()if_else()vector()cross()matvec()quadform()bilinear_form()parse_matvec_component_args()parse_quadform_args()parse_bilinear_form_args()matrix_transpose()matrix_add()
- Module contents
CSEAssignmentCSEPlanCodeGenerationBuilderCodeGenerationBuilder.functionCodeGenerationBuilder.configCodeGenerationBuilder.requestsCodeGenerationBuilder.simplificationCodeGenerationBuilder.functionsCodeGenerationBuilder.with_backend_config()CodeGenerationBuilder.with_simplification()CodeGenerationBuilder.add_primal()CodeGenerationBuilder.add_gradient()CodeGenerationBuilder.add_jacobian()CodeGenerationBuilder.add_vjp()CodeGenerationBuilder.add_joint()CodeGenerationBuilder.add_hessian()CodeGenerationBuilder.add_hvp()CodeGenerationBuilder.for_function()CodeGenerationBuilder.build()
ComposedFunctionComposedFunction.nameComposedFunction.state_inputComposedFunction.input_nameComposedFunction.parameter_nameComposedFunction.simplificationComposedFunction.stepsComposedFunction.finishedComposedFunction.parameter_sizeComposedFunction.stage_countComposedFunction.inputsComposedFunction.outputsComposedFunction.nodesComposedFunction.input_namesComposedFunction.output_namesComposedFunction.then()ComposedFunction.chain()ComposedFunction.repeat()ComposedFunction.finish()ComposedFunction.to_function()ComposedFunction.gradient()ComposedFunction.jacobian()ComposedFunction.joint()ComposedFunction.generate_rust()ComposedFunction.create_rust_project()
ComposedGradientFunctionComposedJacobianFunctionComposedJointFunctionComposedJointFunction.composedComposedJointFunction.nameComposedJointFunction.componentsComposedJointFunction.wrt_indexComposedJointFunction.simplificationComposedJointFunction.to_function()ComposedJointFunction.nodesComposedJointFunction.generate_rust()ComposedJointFunction.create_rust_project()
FunctionFunction.nameFunction.inputsFunction.outputsFunction.input_namesFunction.output_namesFunction.flat_inputsFunction.flat_outputsFunction.nodesFunction.cse()Function.generate_rust()Function.create_rust_project()Function.create_rust_derivative_bundle()Function.jvp()Function.gradient()Function.joint()Function.hvp()Function.vjp()Function.jacobian()Function.jacobian_blocks()Function.vjp_blocks()Function.hessian()Function.hessian_blocks()Function.hvp_blocks()Function.simplify()
FunctionBundleFunctionComposerFunctionCompositionFunctionComposition.nameFunctionComposition.stagesFunctionComposition.input_namesFunctionComposition.output_namesFunctionComposition.nodesFunctionComposition.to_function()FunctionComposition.gradient()FunctionComposition.hessian()FunctionComposition.hvp()FunctionComposition.generate_rust()FunctionComposition.create_rust_project()
RustBackendModeRustBackendConfigRustBackendConfig.backend_modeRustBackendConfig.scalar_typeRustBackendConfig.crate_nameRustBackendConfig.function_nameRustBackendConfig.headerRustBackendConfig.emit_metadata_helpersRustBackendConfig.enable_python_interfaceRustBackendConfig.build_python_interfaceRustBackendConfig.build_crateRustBackendConfig.build_profileRustBackendConfig.additional_dependenciesRustBackendConfig.with_backend_mode()RustBackendConfig.with_scalar_type()RustBackendConfig.with_crate_name()RustBackendConfig.with_function_name()RustBackendConfig.with_header()RustBackendConfig.with_emit_metadata_helpers()RustBackendConfig.with_enable_python_interface()RustBackendConfig.with_build_python_interface()RustBackendConfig.with_build_crate()RustBackendConfig.with_build_profile()RustBackendConfig.with_additional_dependencies()
RustCodegenResultRustCodegenResult.sourceRustCodegenResult.python_nameRustCodegenResult.function_nameRustCodegenResult.workspace_sizeRustCodegenResult.input_namesRustCodegenResult.input_sizesRustCodegenResult.output_namesRustCodegenResult.output_sizesRustCodegenResult.backend_modeRustCodegenResult.scalar_typeRustCodegenResult.math_library
RustDerivativeBundleResultRustMultiFunctionProjectResultRustProjectResultRustPythonInterfaceProjectResultRustPythonInterfaceProjectResult.project_dirRustPythonInterfaceProjectResult.cargo_tomlRustPythonInterfaceProjectResult.pyprojectRustPythonInterfaceProjectResult.readmeRustPythonInterfaceProjectResult.lib_rsRustPythonInterfaceProjectResult.module_nameRustPythonInterfaceProjectResult.low_level_crate_name
RustScalarTypeSXSX.nodeSX.sym()SX.const()SX.opSX.argsSX.nameSX.valueSX.metadataSX.sin()SX.cos()SX.tan()SX.asin()SX.acos()SX.atan()SX.atan2()SX.sinh()SX.cosh()SX.tanh()SX.asinh()SX.acosh()SX.atanh()SX.exp()SX.expm1()SX.log()SX.log1p()SX.sqrt()SX.cbrt()SX.erf()SX.erfc()SX.floor()SX.ceil()SX.round()SX.trunc()SX.fract()SX.signum()SX.hypot()SX.abs()SX.maximum()SX.minimum()
SXNodeSXVectorSXVector.elementsSXVector.sym()SXVector.sin()SXVector.cos()SXVector.tan()SXVector.asin()SXVector.acos()SXVector.atan()SXVector.asinh()SXVector.acosh()SXVector.atanh()SXVector.sinh()SXVector.cosh()SXVector.tanh()SXVector.exp()SXVector.expm1()SXVector.log()SXVector.log1p()SXVector.sqrt()SXVector.cbrt()SXVector.erf()SXVector.erfc()SXVector.floor()SXVector.ceil()SXVector.round()SXVector.trunc()SXVector.fract()SXVector.signum()SXVector.abs()SXVector.dot()SXVector.cross()SXVector.sum()SXVector.prod()SXVector.max()SXVector.min()SXVector.mean()SXVector.norm2()SXVector.norm2sq()SXVector.norm1()SXVector.norm_inf()SXVector.norm_p()SXVector.norm_p_to_p()
SingleShootingBundleSingleShootingGradientFunctionSingleShootingGradientFunction.problemSingleShootingGradientFunction.nameSingleShootingGradientFunction.include_statesSingleShootingGradientFunction.simplificationSingleShootingGradientFunction.to_function()SingleShootingGradientFunction.nodesSingleShootingGradientFunction.input_namesSingleShootingGradientFunction.output_namesSingleShootingGradientFunction.generate_rust()SingleShootingGradientFunction.create_rust_project()
SingleShootingHvpFunctionSingleShootingHvpFunction.problemSingleShootingHvpFunction.nameSingleShootingHvpFunction.include_statesSingleShootingHvpFunction.simplificationSingleShootingHvpFunction.to_function()SingleShootingHvpFunction.nodesSingleShootingHvpFunction.input_namesSingleShootingHvpFunction.output_namesSingleShootingHvpFunction.generate_rust()SingleShootingHvpFunction.create_rust_project()
SingleShootingJointFunctionSingleShootingJointFunction.problemSingleShootingJointFunction.bundleSingleShootingJointFunction.nameSingleShootingJointFunction.simplificationSingleShootingJointFunction.to_function()SingleShootingJointFunction.nodesSingleShootingJointFunction.input_namesSingleShootingJointFunction.output_namesSingleShootingJointFunction.generate_rust()SingleShootingJointFunction.create_rust_project()
SingleShootingPrimalFunctionSingleShootingPrimalFunction.problemSingleShootingPrimalFunction.nameSingleShootingPrimalFunction.include_statesSingleShootingPrimalFunction.simplificationSingleShootingPrimalFunction.to_function()SingleShootingPrimalFunction.nodesSingleShootingPrimalFunction.input_namesSingleShootingPrimalFunction.output_namesSingleShootingPrimalFunction.generate_rust()SingleShootingPrimalFunction.create_rust_project()
SingleShootingProblemSingleShootingProblem.nameSingleShootingProblem.horizonSingleShootingProblem.dynamicsSingleShootingProblem.stage_costSingleShootingProblem.terminal_costSingleShootingProblem.initial_state_nameSingleShootingProblem.control_sequence_nameSingleShootingProblem.parameter_nameSingleShootingProblem.simplificationSingleShootingProblem.stage_penaltySingleShootingProblem.terminal_penaltySingleShootingProblem.penalty_weightSingleShootingProblem.with_horizon()SingleShootingProblem.with_dynamics()SingleShootingProblem.with_stage_cost()SingleShootingProblem.with_terminal_cost()SingleShootingProblem.with_costs()SingleShootingProblem.with_penalties()SingleShootingProblem.with_input_names()SingleShootingProblem.with_simplification()SingleShootingProblem.state_sizeSingleShootingProblem.control_sizeSingleShootingProblem.parameter_sizeSingleShootingProblem.has_runtime_penalty_weightSingleShootingProblem.penalty_weight_nameSingleShootingProblem.input_namesSingleShootingProblem.output_namesSingleShootingProblem.inputsSingleShootingProblem.outputsSingleShootingProblem.nodesSingleShootingProblem.primal()SingleShootingProblem.gradient()SingleShootingProblem.hvp()SingleShootingProblem.joint()SingleShootingProblem.to_function()SingleShootingProblem.stage_total_cost_function()SingleShootingProblem.terminal_total_cost_function()SingleShootingProblem.generate_rust()SingleShootingProblem.create_rust_project()
SquaredDistanceToSetSquaredDistanceToSet.nameSquaredDistanceToSet.with_sq_distance()SquaredDistanceToSet.with_sq_distance_function()SquaredDistanceToSet.with_projection()SquaredDistanceToSet.with_projection_function()SquaredDistanceToSet.with_rust_sq_distance()SquaredDistanceToSet.with_rust_projection()SquaredDistanceToSet.euclidean_ball()SquaredDistanceToSet.infinity_ball()SquaredDistanceToSet.rectangle()SquaredDistanceToSet.second_order_cone()SquaredDistanceToSet.to_function()SquaredDistanceToSet.gradient()SquaredDistanceToSet.jacobian()
acosh()acos()asinh()asin()atan2()atan()atanh()bilinear_form()cbrt()ceil()const()cos()cosh()cross()cse()create_rust_project()create_rust_derivative_bundle()create_multi_function_rust_project()derivative()clear_registered_elementary_functions()erf()erfc()exp()expm1()floor()fract()gradient()get_registered_elementary_function()hessian()hypot()if_else()jacobian()jvp()log()log1p()map_function()reduce_function()matvec()maximum()minimum()quadform()round()register_elementary_function()generate_rust()signum()simplify()sin()sinh()sqrt()tan()tanh()trunc()vector()vjp()zip_function()BatchedFunctionBatchedFunction.functionBatchedFunction.countBatchedFunction.nameBatchedFunction.input_sequence_namesBatchedFunction.simplificationBatchedFunction.functionBatchedFunction.countBatchedFunction.nameBatchedFunction.input_sequence_namesBatchedFunction.simplificationBatchedFunction.input_namesBatchedFunction.output_namesBatchedFunction.nodesBatchedFunction.jacobian()BatchedFunction.to_function()BatchedFunction.generate_rust()BatchedFunction.create_rust_project()
BatchedJacobianFunctionBatchedJacobianFunction.batchedBatchedJacobianFunction.wrt_indexBatchedJacobianFunction.nameBatchedJacobianFunction.simplificationBatchedJacobianFunction.batchedBatchedJacobianFunction.wrt_indexBatchedJacobianFunction.nameBatchedJacobianFunction.simplificationBatchedJacobianFunction.to_function()BatchedJacobianFunction.input_namesBatchedJacobianFunction.output_namesBatchedJacobianFunction.generate_rust()BatchedJacobianFunction.create_rust_project()
ReducedFunctionReducedFunction.functionReducedFunction.countReducedFunction.nameReducedFunction.accumulator_input_nameReducedFunction.input_nameReducedFunction.output_nameReducedFunction.simplificationReducedFunction.functionReducedFunction.countReducedFunction.nameReducedFunction.accumulator_input_nameReducedFunction.input_nameReducedFunction.output_nameReducedFunction.simplificationReducedFunction.input_namesReducedFunction.output_namesReducedFunction.nodesReducedFunction.to_function()ReducedFunction.generate_rust()ReducedFunction.create_rust_project()