firedrake.ml package

Subpackages

• firedrake.ml.jax package
  • Submodules
  • firedrake.ml.jax.fem_operator module
    • FiredrakeJaxOperator
      • FiredrakeJaxOperator.bwd()
      • FiredrakeJaxOperator.forward
      • FiredrakeJaxOperator.fwd()
    • fem_operator()
    • from_jax()
    • to_jax()
  • firedrake.ml.jax.ml_operator module
    • JaxOperator
      • JaxOperator.ufl_operands
    • custom_vjp()
    • ml_operator()
  • Module contents
• firedrake.ml.pytorch package
  • Submodules
  • firedrake.ml.pytorch.fem_operator module
    • FiredrakeTorchOperator
      • FiredrakeTorchOperator.backward()
      • FiredrakeTorchOperator.forward()
    • fem_operator()
    • from_torch()
    • to_torch()
    • torch_operator()
  • firedrake.ml.pytorch.ml_operator module
    • PytorchOperator
      • PytorchOperator.model_output
      • PytorchOperator.torch_grad_enabled
      • PytorchOperator.ufl_operands
    • ml_operator()
    • neuralnet()
  • Module contents

Module contents