Skip to content

Task 06: use TensorCircuit native jaxode (1.50x) - #13

Open
QingyunQian wants to merge 3 commits into
sxzgroup:v2from
QingyunQian:codex/task-06-final-expert-optimization
Open

Task 06: use TensorCircuit native jaxode (1.50x)#13
QingyunQian wants to merge 3 commits into
sxzgroup:v2from
QingyunQian:codex/task-06-final-expert-optimization

Conversation

@QingyunQian

@QingyunQian QingyunQian commented Jul 29, 2026

Copy link
Copy Markdown

Take-home insight

Replacing the Python-side ODE path with TensorCircuit's native jaxode integration is the only material factor; sparse Hamiltonian conversion and tiny setup tweaks do not help.

Measured result

Comparison Runtime Speedup
Public expert 41.4259 s 1.000x
Optimized 27.5366 s 1.50446x

The README keeps one compact ablation table and one figure so the dominant contribution is immediately visible.

Reviewer-facing files

  • optimized_solutions/challenge-06/README.md
  • optimized_solutions/challenge-06/solution_6_native_jaxode.py
  • optimized_solutions/challenge-06/factor-ablation.svg

Full profiles and research chronology remain in hmyuuu/OrbitBreakersExpertBenchmarks.

@QingyunQian QingyunQian changed the title Task 06: publish TensorCircuit native-jaxode optimization (1.50x) Task 06: use TensorCircuit native jaxode (1.50x) Jul 30, 2026
@refraction-ray

Copy link
Copy Markdown
Member

the insight is vague
replace what by jaxode?to me jaxode is much worse than diffrax ode in my experience

@QingyunQian

Copy link
Copy Markdown
Author

the insight is vague replace what by jaxode?to me jaxode is much worse than diffrax ode in my experience

maybe we need to compare the performance of jaxode with diffrax ode on more different situation

@QingyunQian
QingyunQian changed the base branch from main to v2 August 3, 2026 06:07
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.

2 participants