diff --git a/exla/CHANGELOG.md b/exla/CHANGELOG.md
index 68b2b9bcb8..299c4a5141 100644
--- a/exla/CHANGELOG.md
+++ b/exla/CHANGELOG.md
@@ -1,5 +1,17 @@
 # Changelog
 
+## v0.9.0 (2024-09-26)
+
+### Enhancements
+
+  * Overall improvements to the Nx.Defn compiler
+  * Compiled functions now work across BEAM nodes
+  * Add `cache: "path/to/file"` for disk caching JIT/compiled functions
+
+### Bug fixes
+
+  * Use a single thread pool for MLIR contexts
+
 ## v0.8.0 (2024-08-19)
 
   * Add `EXLA.to_mlir_module/2`
diff --git a/exla/mix.exs b/exla/mix.exs
index 2a89d9ee67..d058da0869 100644
--- a/exla/mix.exs
+++ b/exla/mix.exs
@@ -2,7 +2,7 @@ defmodule EXLA.MixProject do
   use Mix.Project
 
   @source_url "https://github.com/elixir-nx/nx"
-  @version "0.8.0"
+  @version "0.9.0"
 
   def project do
     make_args =
@@ -63,8 +63,8 @@ defmodule EXLA.MixProject do
 
   defp deps do
     [
-      # {:nx, "~> 0.8.0"},
-      {:nx, path: "../nx"},
+      {:nx, "~> 0.9.0"},
+      # {:nx, path: "../nx"},
       {:telemetry, "~> 0.4.0 or ~> 1.0"},
       {:xla, "~> 0.8.0", runtime: false},
       {:elixir_make, "~> 0.6", runtime: false},
diff --git a/exla/mix.lock b/exla/mix.lock
index 77b757945d..cb55ce51fc 100644
--- a/exla/mix.lock
+++ b/exla/mix.lock
@@ -10,7 +10,7 @@
   "makeup_erlang": {:hex, :makeup_erlang, "1.0.1", "c7f58c120b2b5aa5fd80d540a89fdf866ed42f1f3994e4fe189abebeab610839", [:mix], [{:makeup, "~> 1.0", [hex: :makeup, repo: "hexpm", optional: false]}], "hexpm", "8a89a1eeccc2d798d6ea15496a6e4870b75e014d1af514b1b71fa33134f57814"},
   "nimble_parsec": {:hex, :nimble_parsec, "1.4.0", "51f9b613ea62cfa97b25ccc2c1b4216e81df970acd8e16e8d1bdc58fef21370d", [:mix], [], "hexpm", "9c565862810fb383e9838c1dd2d7d2c437b3d13b267414ba6af33e50d2d1cf28"},
   "nimble_pool": {:hex, :nimble_pool, "1.0.0", "5eb82705d138f4dd4423f69ceb19ac667b3b492ae570c9f5c900bb3d2f50a847", [:mix], [], "hexpm", "80be3b882d2d351882256087078e1b1952a28bf98d0a287be87e4a24a710b67a"},
-  "nx": {:hex, :nx, "0.8.0", "81d801773cbcee654b8f6a41ccb7c2716d25073f2d64fec3d62950c9db98cf99", [:mix], [{:complex, "~> 0.5", [hex: :complex, repo: "hexpm", optional: false]}, {:telemetry, "~> 0.4.0 or ~> 1.0", [hex: :telemetry, repo: "hexpm", optional: false]}], "hexpm", "e61943143f1719dcceb8c5004286b719f3ab230058eafeea39799a4da3f0d754"},
+  "nx": {:hex, :nx, "0.9.0", "03a622a27d93eaaa2d24ff9b812d9f675cc04eb0340ca3dd065674f3642867d3", [:mix], [{:complex, "~> 0.5", [hex: :complex, repo: "hexpm", optional: false]}, {:telemetry, "~> 0.4.0 or ~> 1.0", [hex: :telemetry, repo: "hexpm", optional: false]}], "hexpm", "3810a5a90db0654b6e538430c0fb473a22bfc11b3d02ea7834db493cf3f56153"},
   "statistex": {:hex, :statistex, "1.0.0", "f3dc93f3c0c6c92e5f291704cf62b99b553253d7969e9a5fa713e5481cd858a5", [:mix], [], "hexpm", "ff9d8bee7035028ab4742ff52fc80a2aa35cece833cf5319009b52f1b5a86c27"},
   "telemetry": {:hex, :telemetry, "1.3.0", "fedebbae410d715cf8e7062c96a1ef32ec22e764197f70cda73d82778d61e7a2", [:rebar3], [], "hexpm", "7015fc8919dbe63764f4b4b87a95b7c0996bd539e0d499be6ec9d7f3875b79e6"},
   "xla": {:hex, :xla, "0.8.0", "fef314d085dd3ee16a0816c095239938f80769150e15db16dfaa435553d7cb16", [:make, :mix], [{:elixir_make, "~> 0.4", [hex: :elixir_make, repo: "hexpm", optional: false]}], "hexpm", "739c61c8d93b97e12ba0369d10e76130224c208f1a76ad293e3581f056833e57"},
diff --git a/nx/CHANGELOG.md b/nx/CHANGELOG.md
index 534090c582..d7cb2e7212 100644
--- a/nx/CHANGELOG.md
+++ b/nx/CHANGELOG.md
@@ -1,5 +1,23 @@
 # Changelog
 
+## v0.9.0 (2024-09-26)
+
+### Enhancements
+
+  * [Nx] Add 8-bit Floating Point numerical type
+  * [Nx] Added `s4` and `s2` quantized integer types
+  * [Nx] Add quantized int types (s2, s4, u2, u4)
+
+### Bug fixes
+
+  * [Nx.LinAlg] Minor range slicing fixes on QR decomposition
+  * [Nx] Nx.Defn.Grad now supports more vectorization cases
+
+### Deprecations and incompatibilities
+
+  * [Nx] Default integer type is now `s32`
+  * [Nx] Interface breaking changes for `Nx.to_pointer` and `Nx.from_pointer`
+
 ## v0.8.0 (2024-08-19)
 
 ### Enhancements
diff --git a/nx/mix.exs b/nx/mix.exs
index ec3b6b08a0..d2cd6d31a8 100644
--- a/nx/mix.exs
+++ b/nx/mix.exs
@@ -2,7 +2,7 @@ defmodule Nx.MixProject do
   use Mix.Project
 
   @source_url "https://github.com/elixir-nx/nx"
-  @version "0.8.0"
+  @version "0.9.0"
 
   def project do
     [
diff --git a/torchx/CHANGELOG.md b/torchx/CHANGELOG.md
index 9bbd434ffd..c9bd06e330 100644
--- a/torchx/CHANGELOG.md
+++ b/torchx/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## v0.9.0 (2024-09-26)
+
+  * Update to latest Nx
+
 ## v0.8.0 (2024-08-19)
 
   * Update to latest Nx
diff --git a/torchx/mix.exs b/torchx/mix.exs
index b1d2347346..7f85943726 100644
--- a/torchx/mix.exs
+++ b/torchx/mix.exs
@@ -2,7 +2,7 @@ defmodule Torchx.MixProject do
   use Mix.Project
 
   @source_url "https://github.com/elixir-nx/nx"
-  @version "0.8.0"
+  @version "0.9.0"
 
   @libtorch_compilers [:torchx, :cmake]
 
@@ -41,8 +41,8 @@ defmodule Torchx.MixProject do
 
   defp deps do
     [
-      # {:nx, "~> 0.8.0"},
-      {:nx, path: "../nx"},
+      {:nx, "~> 0.9.0"},
+      # {:nx, path: "../nx"},
       {:ex_doc, "~> 0.29", only: :docs}
     ]
   end
diff --git a/torchx/mix.lock b/torchx/mix.lock
index bcd6f89157..b3a93517e0 100644
--- a/torchx/mix.lock
+++ b/torchx/mix.lock
@@ -6,6 +6,6 @@
   "makeup_elixir": {:hex, :makeup_elixir, "0.16.2", "627e84b8e8bf22e60a2579dad15067c755531fea049ae26ef1020cad58fe9578", [:mix], [{:makeup, "~> 1.0", [hex: :makeup, repo: "hexpm", optional: false]}, {:nimble_parsec, "~> 1.2.3 or ~> 1.3", [hex: :nimble_parsec, repo: "hexpm", optional: false]}], "hexpm", "41193978704763f6bbe6cc2758b84909e62984c7752b3784bd3c218bb341706b"},
   "makeup_erlang": {:hex, :makeup_erlang, "1.0.1", "c7f58c120b2b5aa5fd80d540a89fdf866ed42f1f3994e4fe189abebeab610839", [:mix], [{:makeup, "~> 1.0", [hex: :makeup, repo: "hexpm", optional: false]}], "hexpm", "8a89a1eeccc2d798d6ea15496a6e4870b75e014d1af514b1b71fa33134f57814"},
   "nimble_parsec": {:hex, :nimble_parsec, "1.4.0", "51f9b613ea62cfa97b25ccc2c1b4216e81df970acd8e16e8d1bdc58fef21370d", [:mix], [], "hexpm", "9c565862810fb383e9838c1dd2d7d2c437b3d13b267414ba6af33e50d2d1cf28"},
-  "nx": {:hex, :nx, "0.8.0", "81d801773cbcee654b8f6a41ccb7c2716d25073f2d64fec3d62950c9db98cf99", [:mix], [{:complex, "~> 0.5", [hex: :complex, repo: "hexpm", optional: false]}, {:telemetry, "~> 0.4.0 or ~> 1.0", [hex: :telemetry, repo: "hexpm", optional: false]}], "hexpm", "e61943143f1719dcceb8c5004286b719f3ab230058eafeea39799a4da3f0d754"},
+  "nx": {:hex, :nx, "0.9.0", "03a622a27d93eaaa2d24ff9b812d9f675cc04eb0340ca3dd065674f3642867d3", [:mix], [{:complex, "~> 0.5", [hex: :complex, repo: "hexpm", optional: false]}, {:telemetry, "~> 0.4.0 or ~> 1.0", [hex: :telemetry, repo: "hexpm", optional: false]}], "hexpm", "3810a5a90db0654b6e538430c0fb473a22bfc11b3d02ea7834db493cf3f56153"},
   "telemetry": {:hex, :telemetry, "1.3.0", "fedebbae410d715cf8e7062c96a1ef32ec22e764197f70cda73d82778d61e7a2", [:rebar3], [], "hexpm", "7015fc8919dbe63764f4b4b87a95b7c0996bd539e0d499be6ec9d7f3875b79e6"},
 }