• X
    Enhance vjp/jvp/Jacobian/Hessian API for supporting dynamic, static graph and... · 9e764d82
    Xiaoxu Chen 提交于
    Enhance vjp/jvp/Jacobian/Hessian API for supporting dynamic, static graph and batched, unbatched mode (#40692)
    
    * modify vjp/jvp for both dynamic and static graph
    
    * enforce jacobian class for supporting first/last batch
    
    * add unittest for jvp, jacobian withlast batch, jacobian with first batch
    
    * fix the incorrect shape when multi-index Jacobian
    
    * enforce Hessian class for supporting dynamic graph
    
    * add Hessian class unittest
    
    * bugfix, jvp double_backward_trick zeros_like return stop_gradient=True in static graph
    
    * add API beta warnnings
    
    * add white_list for cuda11.x ci windows.
    
    * optimize some code snippets and documments
    
    * set unittest timeout to 100 seconds
    
    * move vjp,jvp,Jacobian,Hessian to incubate
    
    * fix vjp,vjp import path of sample code
    
    * fix code style error of augtograd/__init__ file
    9e764d82
__init__.py 1.5 KB