diff --git a/docs/conf.py b/docs/conf.py
index 4bf0da5..f2c6669 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/__init__.py b/last/__init__.py
index 3e647ef..575eb4d 100644
--- a/last/__init__.py
+++ b/last/__init__.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,4 +21,4 @@
 from last.lattices import RecognitionLattice
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
-__version__ = '0.0.3'
+__version__ = '0.0.4'
diff --git a/last/alignments.py b/last/alignments.py
index 384b317..57b85de 100644
--- a/last/alignments.py
+++ b/last/alignments.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/contexts.py b/last/contexts.py
index 683a8d0..465cf26 100644
--- a/last/contexts.py
+++ b/last/contexts.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/lattices.py b/last/lattices.py
index 5a9bcfa..cd2e8a0 100644
--- a/last/lattices.py
+++ b/last/lattices.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/scatter_reduce.py b/last/scatter_reduce.py
index 38fdbf2..a1a86e8 100644
--- a/last/scatter_reduce.py
+++ b/last/scatter_reduce.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/semirings.py b/last/semirings.py
index d85bad4..c0cef33 100644
--- a/last/semirings.py
+++ b/last/semirings.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/last/weight_fns.py b/last/weight_fns.py
index 1f2041f..2ece7b7 100644
--- a/last/weight_fns.py
+++ b/last/weight_fns.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/pyproject.toml b/pyproject.toml
index 4c579f2..c75eed5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,9 +16,9 @@ keywords = ["speech", "jax"]
 
 # pip dependencies of the project
 dependencies = [
-  "einops>=0.5.0",
-  "flax>=0.7.0",
-  "jax>=0.3.21",
+  "einops>=0.7.0",
+  "flax>=0.8.2",
+  "jax>=0.4.25",
 ]
 
 # This is set automatically by flit using `last.__version__`
diff --git a/tests/alignments_test.py b/tests/alignments_test.py
index 6771507..011d693 100644
--- a/tests/alignments_test.py
+++ b/tests/alignments_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/contexts_test.py b/tests/contexts_test.py
index 5b96ae1..34aac8f 100644
--- a/tests/contexts_test.py
+++ b/tests/contexts_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/last_test.py b/tests/last_test.py
index 35bbfea..ee68897 100644
--- a/tests/last_test.py
+++ b/tests/last_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/lattices_test.py b/tests/lattices_test.py
index 212a00f..1fad6be 100644
--- a/tests/lattices_test.py
+++ b/tests/lattices_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/semirings_test.py b/tests/semirings_test.py
index 91162e7..19d7da4 100644
--- a/tests/semirings_test.py
+++ b/tests/semirings_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/toy_tasks_test.py b/tests/toy_tasks_test.py
index 4a4969e..642160b 100644
--- a/tests/toy_tasks_test.py
+++ b/tests/toy_tasks_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/tests/weight_fns_test.py b/tests/weight_fns_test.py
index 976b22e..f894d42 100644
--- a/tests/weight_fns_test.py
+++ b/tests/weight_fns_test.py
@@ -1,4 +1,4 @@
-# Copyright 2023 The LAST Authors.
+# Copyright 2024 The LAST Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.