Skip to content

Sparse MatMul Support in TorchJD for Seamless compatibility with torch.vmap() #388

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
EmileAydar wants to merge 3 commits into
base: main
Choose a base branch
from

[pre-commit.ci] auto fixes from pre-commit.com hooks

a0d68e3
Select commit
Loading
Failed to load commit list.
Open

Sparse MatMul Support in TorchJD for Seamless compatibility with torch.vmap() #388

[pre-commit.ci] auto fixes from pre-commit.com hooks
a0d68e3
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch failed Jun 12, 2025 in 0s

98.01% of diff hit (target 100.00%)

View this Pull Request on Codecov

98.01% of diff hit (target 100.00%)

Annotations

Check warning on line 35 in src/torchjd/sparse/_patch.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

src/torchjd/sparse/_patch.py#L35 

Added line #L35 was not covered by tests

Check warning on line 76 in src/torchjd/sparse/_patch.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

src/torchjd/sparse/_patch.py#L76 

Added line #L76 was not covered by tests
View more details on Codecov