Skip to content

Optimize memory usage - #2

Open
Anya497 wants to merge 10 commits into
mainfrom
optimize_memory_usage
Open

Optimize memory usage#2
Anya497 wants to merge 10 commits into
mainfrom
optimize_memory_usage

Conversation

@Anya497

@Anya497 Anya497 commented Apr 25, 2026

Copy link
Copy Markdown
Owner

This PR

  • optimizes GradDFT training by more than 10-fold escaping intermediate representation in einsum operator;
  • tries to fix all tests;
  • removes unfixable tests which also fail in main now.

@Anya497
Anya497 force-pushed the optimize_memory_usage branch from 3c86030 to 4fb5aa9 Compare April 25, 2026 18:01
@Anya497
Anya497 force-pushed the optimize_memory_usage branch 2 times, most recently from 61ca8ac to 516885a Compare April 25, 2026 18:31
@Anya497
Anya497 force-pushed the optimize_memory_usage branch 2 times, most recently from 7eaf76d to 0749356 Compare July 2, 2026 14:08
Use integers.

Modify dependencies usage.

Fix issue with as_dict.

Try remove add_signature calling.

Fix warnings and error with clip.
@Anya497
Anya497 force-pushed the optimize_memory_usage branch from a7cbdbc to b279797 Compare August 22, 2026 16:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant