Here is a gallery of tutorials and examples to get you started with ScimBa.

Tutorials

Each tutorial is a single Jupyter notebook that helps you get started with Scimba.

Basics of scimba_jax

Solve pre-defined stationary and time-dependent models.

Scimba basics I: approximation of the solution of pre-defined physical models

Strong Boundary Conditions

Implementation of strong boundary and initial conditions.

Strong boundary conditions

Mixed Boundary Conditions

How to apply mixed boundary conditions.

Mixed Boundary Conditions with PINNs

Domains and Samplers

How to define domains and samplers.

Domains and Samplers

Save and load trained models.

How to save and load objects.

Save and load PINNs and other scimba_jax objects

Custom Physical models

How to define custom physical models.

Defining a physical model

Examples

A complete list of JAX script examples is available in the examples/examples_jax/ directory of the repository.