diff --git a/pyrad/nerf/field_modules/encoding.py b/pyrad/nerf/field_modules/encoding.py
index 1632030ea..3b65039b1 100644
--- a/pyrad/nerf/field_modules/encoding.py
+++ b/pyrad/nerf/field_modules/encoding.py
@@ -139,6 +139,7 @@ class RFFEncoding(Encoding):
             raise ValueError("RFF encoding scale should be greater than zero")
         self.scale = scale
         self.b_matrix = torch.normal(mean=0, std=self.scale, size=(self.in_dim, self.num_frequencies))
+        self.register_buffer(name="rff_scalings", tensor=self.b_matrix)
 
     def get_out_dim(self) -> int:
         return self.num_frequencies * 2