eqx.filter_vmap failing to respect in_axes kwarg

equinox_inaxes.py

import equinox as eqx
import jax
import numpy as np


def func(x):
  return x + x

v_func_jax = jax.vmap(func, in_axes=0)
v_func_eqx = eqx.filter_vmap(func, in_axes=0)

x = np.array([[1, 2], [3, 4]], dtype=np.float64)
exp = np.array([[2, 4], [6, 8]], dtype=np.float64)

y_j = v_func_jax(x)
np.testing.assert_array_equal(y_j, exp)
y_e = v_func_eqx(x)
# Traceback (most recent call last):
#   File ".../equinox_inaxes.py", line 18, in <module>
#     y_e = v_func_eqx(x)
#   File ".../site-packages/equinox/vmap_pmap.py", line 146, in __call__
#     vmapd, nonvmapd = jax.vmap(
# TypeError: jax._src.api.vmap() got multiple values for keyword argument 'in_axes'