Bumped jax-triton version along with JAX/Triton requirements

PiperOrigin-RevId: 753944885

Author: superbobry

jax_triton/version.py

@@ -12,5 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version_info__ = (0, 2, 0)
+__version_info__ = (0, 3, 0)
 __version__ = ".".join(str(v) for v in __version_info__)

pyproject.toml

@@ -6,8 +6,8 @@ readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
   "absl-py>=1.4.0",
-  "jax>=0.4.34",
-  "triton>=3.1",
+  "jax>=0.6.0",
+  "triton>=3.3",
 ]
 
 [project.optional-dependencies]