diff --git a/arraycontext/impl/jax/__init__.py b/arraycontext/impl/jax/__init__.py index a70cbaa2..d767a083 100644 --- a/arraycontext/impl/jax/__init__.py +++ b/arraycontext/impl/jax/__init__.py @@ -121,7 +121,7 @@ def tag(self, tags: ToTagSetConvertible, array): return array def tag_axis(self, iaxis, tags: ToTagSetConvertible, array): - # TODO: See `jax.experiemental.maps.xmap`, probably that should be useful? + # TODO: See `jax.experimental.maps.xmap`, probably that should be useful? return array def call_loopy(self, t_unit, **kwargs):