All pull requests

Pull requests

• Add --configs flag to run_and_save_all_benchmarks.py to allow selective runs.

  #2494·copybara-service[bot] opened Sep 24, 2026 · Closed Sep 24, 2026Bot·

  ·

• Add dedicated turbulent_transport child node in output DataTree.

  #2490·copybara-service[bot] opened Sep 21, 2026 · Closed Sep 24, 2026Bot·

  ·

• Replace jax.lax.cond with Python if for Pereverzev transport calculation.

  #2489·copybara-service[bot] opened Sep 21, 2026 · Closed Sep 24, 2026Bot·

  ·

• Add Counter type alias in jit_run_loop.

  #2486·copybara-service[bot] opened Sep 21, 2026 · Closed Sep 21, 2026Bot·

  ·

• Vectorize _build_smoothing_matrix masking and use jnp.where in smooth_single_coeff

  #2484·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 24, 2026Bot·

  ·

• Make use_pereverzev static in transport_coefficients_builder

  #2483·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 22, 2026Bot·

  ·

• [TORAX] Support lazy callable cond in error_if to avoid tracing disabled checks

  #2482·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 24, 2026Bot·

  ·

• Replace .at[...].add scatters with broadcast identity in theta_method_matrix_equation

  #2481·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 24, 2026Bot·

  ·

• Use Patankar single-power A(|Pe|) formulation in convection_terms.peclet_to_alpha

  #2480·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 24, 2026Bot·

  ·

• [TORAX] Pre-add 1D channel TriDiagonals and remove scatter-adds in discrete_system.calc_c

  #2479·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 25, 2026Bot·

  ·

• Use closed-form 1x1-4x4 block matrix inversion in BlockTriDiagonal.thomas_solve

  #2477·copybara-service[bot] opened Sep 20, 2026 · Closed Sep 24, 2026Bot·

  ·

• Cache CellVariable grid properties and face interpolations via @cached_property.

  #2476·copybara-service[bot] opened Sep 19, 2026 · Closed Sep 24, 2026Bot·

  ·

• Add GACODE NEO surrogate neoclassical transport and bootstrap current models to TORAX.

  #2470·copybara-service[bot] opened Sep 17, 2026 · Closed Sep 17, 2026Bot·

  ·
  1

• Add T_e_right_bc and T_i_right_bc to EdgeModelOutputs.

  #2469·copybara-service[bot] opened Sep 16, 2026 · Closed Sep 18, 2026Bot·

  ·

• Deduplicate ExtendedLengyelOutputs construction and refine SOL state types.

  #2468·copybara-service[bot] opened Sep 16, 2026 · Closed Sep 18, 2026Bot·

  ·

• suppress pyrefly errors related to JAX scalar types

  #2467·copybara-service[bot] opened Sep 16, 2026 · Closed Sep 16, 2026Bot·

  ·

• Switch to explicit compiler options in the JITted function rather than relying on global flags.

  #2461·copybara-service[bot] opened Sep 14, 2026 · Closed Sep 16, 2026Bot·

  ·

• Remove dependency on typing_extensions.

  #2460·copybara-service[bot] opened Sep 14, 2026 · Closed Sep 23, 2026Bot·

  ·

• Add BetaPoloidalPrimeIBC model for L-mode edge kinetic profile parameterization

  #2455·copybara-service[bot] opened Sep 12, 2026 · Closed Sep 19, 2026Bot·

  ·

• Split profile_conditions into Config and RuntimeParams modules.

  #2453·copybara-service[bot] opened Sep 11, 2026 · Closed Sep 19, 2026Bot·

  ·

• Add JIT annotations to speed up tracing/lowering.

  #2452·copybara-service[bot] opened Sep 11, 2026 · Closed Sep 18, 2026Bot·

  ·

• Add polymorphic InternalBoundaryConditionModel hierarchy and dynamic solver integration

  #2451·copybara-service[bot] opened Sep 10, 2026 · Closed Sep 19, 2026Bot·

  ·

• Move internal boundary conditions to top-level of ToraxConfig.

  #2448·copybara-service[bot] opened Sep 10, 2026 · Closed Sep 11, 2026Bot·

  ·

• Suppress new Pyrefly findings before upgrading to v1.2.0

  #2447·copybara-service[bot] opened Sep 9, 2026 · Closed Sep 9, 2026Bot·

  ·

• Expose max_linesearch_steps in Newton-Raphson config.

  #2446·copybara-service[bot] opened Sep 9, 2026 · Closed Sep 10, 2026Bot·

  ·