PhysicsNeMo Layers#
PhysicsNeMo Layers
- Activations
- Attention and Transformers
AttentionOpEarthAttention2DEarthAttention3DUNetAttentionDecoderLayerEncoderLayerFuserLayerSwinTransformerTransformer2DBlockTransformer3DBlockDecoderOnlyLayerTransformerDecoderPhysicsAttentionBasePhysicsAttentionIrregularMeshPhysicsAttentionStructuredMesh2DPhysicsAttentionStructuredMesh3D
- Convolutional Layers
- Embeddings
- Fourier, FFT, and Spectral Layers
FourierFilterFourierLayerFourierMLPGaborFilterfourier_encode()SpectralConv1dSpectralConv2dSpectralConv3dSpectralConv4dcalc_latent_derivatives()first_order_pino_grads()fourier_derivatives()second_order_pino_grads()OnnxIrfftOnnxIrfft2OnnxRfftOnnxRfft2imag()irfft()irfft2()real()rfft()rfft2()view_as_complex()AFNO2DLayerAFNOMlpAFNOPatchEmbedModAFNO2DLayerModAFNOMlpPatchEmbedScaleShiftMlpFNO1DEncoderFNO2DEncoderFNO3DEncoderFNO4DEncoder
- Fully Connected and MLP Layers
- Normalization
- Resampling and Interpolation
- Regularization
- Specialized Layers
- Graph and Geometry Layers