Warning

You are reading a version of the website built against the unstable main branch. This content is liable to change without notice and may be inappropriate for your use case. You can find the documentation for the current stable release here.

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
    • 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