Skip to content

Add JIT annotations to remaining JAX functions in Torax. - #2458

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_981031710
Open

copybara-service[bot] wants to merge 1 commit into
mainfrom
test_981031710

Conversation

@copybara-service

@copybara-service copybara-service Bot commented Sep 14, 2026 •

Copy link
Copy Markdown

Add JIT annotations to remaining JAX functions in Torax.

@copybara-service
copybara-service Bot force-pushed the test_981031710 branch 3 times, most recently from 755db12 to c708acf Compare September 14, 2026 14:01
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