fix(ops): export split_with_sizes (#315)

Author: 103yiran

src/flag_gems/ops/__init__.py

@@ -784,6 +784,7 @@
     special_spherical_bessel_j0_,
 )
 from flag_gems.ops.special_xlog1py import special_xlog1py
+from flag_gems.ops.split_with_sizes import split_with_sizes
 from flag_gems.ops.split_with_sizes_copy import split_with_sizes_copy
 from flag_gems.ops.sqrt import sqrt, sqrt_
 from flag_gems.ops.square import square, square_, square_out
@@ -1726,6 +1727,7 @@
     "special_spherical_bessel_j0",
     "special_spherical_bessel_j0_",
     "special_softmax",
+    "split_with_sizes",
     "split_with_sizes_copy",
     "sqrt",
     "sqrt_",