Skip to content
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

Add jac_chunk_size keyword argument to ObjectiveFunction to reduce memory usage of forward mode Jacobian calculation #1052

Merged
merged 74 commits into from
Sep 26, 2024

Merge branch 'master' into dp/jacobian-batched-vmap

3e99510
Select commit
Loading
Failed to load commit list.
Merged

Add jac_chunk_size keyword argument to ObjectiveFunction to reduce memory usage of forward mode Jacobian calculation #1052

Merge branch 'master' into dp/jacobian-batched-vmap
3e99510
Select commit
Loading
Failed to load commit list.
Codecov / codecov/patch failed Sep 26, 2024 in 0s

89.44% of diff hit (target 95.43%)

View this Pull Request on Codecov

89.44% of diff hit (target 95.43%)

Annotations

Check warning on line 11 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L11 

Added line #L11 was not covered by tests

Check warning on line 46 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L46 

Added line #L46 was not covered by tests

Check warning on line 49 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L49 

Added line #L49 was not covered by tests

Check warning on line 53 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L53 

Added line #L53 was not covered by tests

Check warning on line 155 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L155 

Added line #L155 was not covered by tests

Check warning on line 158 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L158 

Added line #L158 was not covered by tests

Check warning on line 164 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L164 

Added line #L164 was not covered by tests

Check warning on line 167 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L167 

Added line #L167 was not covered by tests

Check warning on line 237 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L237 

Added line #L237 was not covered by tests

Check warning on line 242 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L242 

Added line #L242 was not covered by tests

Check warning on line 256 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L255-L256 

Added lines #L255 - L256 were not covered by tests

Check warning on line 261 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L260-L261 

Added lines #L260 - L261 were not covered by tests

Check warning on line 265 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L264-L265 

Added lines #L264 - L265 were not covered by tests

Check warning on line 318 in desc/batching.py

See this annotation in the file changed.

@codecov codecov / codecov/patch

desc/batching.py#L318 

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