jittor.optim
=====================

这里是Jittor的优化器模块的API文档，您可以通过`from jittor import optim`来获取该模块。

```eval_rst
.. automodule:: jittor.optim
   :members:
   :undoc-members: add_param_group, step, get_lr, find_grad, load_state_dict, state_dict, zero_grad, get_gamma, better, 
```

以下是Jittor的学习率调度模块的API文档，学习率调度模块需要配合优化器使用，您可以通过`from jittor import lr_scheduler`来获取该模块。

```eval_rst
.. automodule:: jittor.lr_scheduler
   :members:
   :undoc-members: update_lr
```