Re-fix docstrings of autojit functions