[example] fused_linear_jsd

[v0i0]

No description provided.

[jansel]

Add a test and __main__ for this similar to the other examples?

[yf225]

thanks @v0i0 !

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.

[yf225]

@v0i0 the internal error on expected string mismatch is weird... if a rebase doesn't fix it, we could gate it with not is_fbcode() (can add it to _testing.py):

def is_fbcode() -> bool:
    return not hasattr(torch.version, "git_version")

[oulgen]

@v0i0 the internal error on expected string mismatch is weird... if a rebase doesn't fix it, we could gate it with not is_fbcode() (can add it to _testing.py):

def is_fbcode() -> bool:
    return not hasattr(torch.version, "git_version")

you dont need to define is_fbcode again, just import it from pytorch like

from torch._environment import is_fbcode

[v0i0]

@yf225 will rebase again, looks like that's needed anyways. don't see why it would otherwise fail...

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.

[facebook-github-bot]

@v0i0 has imported this pull request. If you are a Meta employee, you can view this in D82675849.