All modules for which code is available

• sciai.architecture.activation
• sciai.architecture.basic_block
• sciai.architecture.neural_operators.fno1d
• sciai.architecture.neural_operators.fno2d
• sciai.architecture.neural_operators.fno3d
• sciai.architecture.neural_operators.pdenet
• sciai.architecture.transformer.vit
• sciai.common.dataset
• sciai.common.initializer
• sciai.common.optimizer
• sciai.common.train_cell
• sciai.context.context
• sciai.operators.derivatives
• sciai.operators.jacobian_weights
• sciai.utils.check_utils
• sciai.utils.file_utils
• sciai.utils.log_utils
• sciai.utils.ms_utils
• sciai.utils.plot_utils
• sciai.utils.python_utils
• sciai.utils.time_utils