PhysicsNeMo NN

The physicsnemo.nn package provides layers and stateless functionals for building neural networks, aligned with the torch.nn and torch.nn.functional patterns.

• Layers (modules): PhysicsNeMo Layers

• Functionals (stateless ops): PhysicsNeMo Functionals