Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
neuraloperator / __init__.py
Size: Mime:
__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,
)