Using optimization at inference time is a foundational concept of Energy-Based Models (EBM) and Objective-Driven AI architectures (ODAI).
When the variables to be inferred are continuous, it makes sense to use gradient-based optimization.
A good instance of ODAI is world model-based systems that use gradient-based optimization for planning.