diff --git a/src/julax/base.py b/src/julax/base.py
index 792f132..4192573 100644
--- a/src/julax/base.py
+++ b/src/julax/base.py
@@ -31,3 +31,8 @@ def items(self):
 
     def __hash__(self):
         return hash(frozenset(self.root.items()))
+
+    def __eq__(self, other):
+        if isinstance(other, FrozenDict):
+            return self.root == other.root
+        return self.root == other