thunder.dynamo.thunderfx

thunder.dynamo.thunderfx(fn, /, **kwargs)

Compiles a callable (function or model) by using Thunder as the backend of torch.compile().

Parameters:

• fn (Callable) – A Module or a function to compile.

• fn (Callable) –

Keyword Arguments:

**kwargs – a dictionary of options to pass to torch.compile() and thunder.jit().

Return type:

ThunderFXCompiledObject

Returns:

The compiled callable