[MASTER] extension-pkg-whitelist=flash_attn_2_cuda, torch, transformer_engine_torch, transformer_engine_jax disable=too-many-locals, too-few-public-methods, too-many-public-methods, too-many-positional-arguments, invalid-name, too-many-arguments, abstract-method, arguments-differ, too-many-instance-attributes, unsubscriptable-object, import-outside-toplevel, too-many-statements, import-error, too-many-lines, use-maxsplit-arg, protected-access, pointless-string-statement, cyclic-import, duplicate-code, no-member, attribute-defined-outside-init, global-statement, too-many-branches, global-variable-not-assigned, redefined-argument-from-local, line-too-long, too-many-return-statements, too-many-nested-blocks [TYPECHECK] ignored-modules=torch ignored-classes=torch