Repository URL to install this package:
|
Version:
2.0.0 ▾
|
neuraloperator
/
__init__.py
|
|---|
__version__ = "2.0.0"
from .models import TFNO, FNO, get_model
from .data import datasets, transforms
from . import mpu
from .training import Trainer
from .losses import (
LpLoss,
H1Loss,
BurgersEqnLoss,
ICLoss,
WeightedSumLoss,
Aggregator,
Relobralo,
SoftAdapt,
FourierDiff,
non_uniform_fd,
FiniteDiff,
)