jaxmod

• jaxmod package
  • Submodules
  • jaxmod.constants module
    • AVOGADRO
    • GAS_CONSTANT
    • GAS_CONSTANT_BAR
    • GRAVITATIONAL_CONSTANT
    • ATMOSPHERE
    • BOLTZMANN_CONSTANT
    • BOLTZMANN_CONSTANT_BAR
    • EARTH_MASS
    • OCEAN_MOLES
    • OCEAN_MASS_H2
    • OCEAN_MASS_H2O
  • jaxmod.solvers module
    • POSTCHECK_TOLERANCE
    • RootFindParameters
      • RootFindParameters.solver
      • RootFindParameters.atol
      • RootFindParameters.rtol
      • RootFindParameters.linear_solver
      • RootFindParameters.norm()
      • RootFindParameters.throw
      • RootFindParameters.max_steps
      • RootFindParameters.jac
      • RootFindParameters.get_solver_instance()
    • MultiAttemptSolution
      • MultiAttemptSolution.solution
      • MultiAttemptSolution.attempts
      • MultiAttemptSolution.aux
      • MultiAttemptSolution.batch_shape
      • MultiAttemptSolution.converged
      • MultiAttemptSolution.num_steps
      • MultiAttemptSolution.result
      • MultiAttemptSolution.value
      • MultiAttemptSolution.solver_success
      • MultiAttemptSolution.state
      • MultiAttemptSolution.stats
      • MultiAttemptSolution.asdict()
    • max_norm()
    • expand_mask()
    • make_batch_retry_solver()
  • jaxmod.type_aliases module
    • NpArray
    • NpBool
    • NpFloat
    • NpInt
    • Scalar
    • OptxSolver
  • jaxmod.units module
    • UnitConversion
      • UnitConversion.atmosphere_to_bar
      • UnitConversion.bar_to_Pa
      • UnitConversion.bar_to_MPa
      • UnitConversion.bar_to_GPa
      • UnitConversion.Pa_to_bar
      • UnitConversion.MPa_to_bar
      • UnitConversion.GPa_to_bar
      • UnitConversion.fraction_to_ppm
      • UnitConversion.ppm_to_fraction
      • UnitConversion.ppm_to_percent
      • UnitConversion.percent_to_ppm
      • UnitConversion.g_to_kg
      • UnitConversion.cm3_to_m3
      • UnitConversion.m3_to_cm3
      • UnitConversion.litre_to_m3
      • UnitConversion.m3_bar_to_J
      • UnitConversion.J_to_m3_bar
  • jaxmod.utils module
    • as_j64()
    • get_batch_axis()
    • get_batch_size()
    • is_hashable()
    • is_jax_array()
    • partial_rref()
    • power_law()
    • safe_exp()
    • to_hashable()
    • to_native_floats()
    • vmap_axes_spec()
  • Module contents
    • MAX_EXP_INPUT
    • MIN_EXP_INPUT
    • simple_formatter()
    • debug_logger()