From e0b6233fa14b518d61c5369b15a9d4d4dcd76bb5 Mon Sep 17 00:00:00 2001
From: paugier <pierre.augier@univ-grenoble-alpes.fr>
Date: Thu, 6 Jun 2024 14:28:00 +0200
Subject: [PATCH] Prepare 0.7.0 (version bump + release notes + doc)

---
 CHANGES.md                         | 5 +++++
 pyproject.toml                     | 2 +-
 src/transonic/backends/__init__.py | 1 +
 src/transonic/backends/jax.py      | 6 +++++-
 4 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/CHANGES.md b/CHANGES.md
index 89c1352..b1d6d36 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -3,6 +3,10 @@
 See also the
 [unreleased changes](https://foss.heptapod.net/fluiddyn/transonic/-/compare/0.6.4...branch%2Fdefault).
 
+## [0.7.0] (2024-06-06)
+
+- Add a backend for [JAX](https://github.com/google/jax) ({mod}`transonic.backends.jax`).
+
 ## [0.6.4] (2024-03-29)
 
 - The `__transonic__` attribute is now a `str` (should avoid
@@ -270,3 +274,4 @@ See also the
 [0.6.2]: https://foss.heptapod.net/fluiddyn/transonic/-/compare/0.6.1...0.6.2
 [0.6.3]: https://foss.heptapod.net/fluiddyn/transonic/-/compare/0.6.2...0.6.3
 [0.6.4]: https://foss.heptapod.net/fluiddyn/transonic/-/compare/0.6.3...0.6.4
+[0.7.0]: https://foss.heptapod.net/fluiddyn/transonic/-/compare/0.6.4...0.7.0
diff --git a/pyproject.toml b/pyproject.toml
index 1cbf1aa..93253f6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "transonic"
-version = "0.6.4"
+version = "0.7.0"
 description = "Make your Python code fly at transonic speeds!"
 authors = [
     { name = "Pierre Augier", email = "pierre.augier@univ-grenoble-alpes.fr" },
diff --git a/src/transonic/backends/__init__.py b/src/transonic/backends/__init__.py
index beb84ed..6fec605 100644
--- a/src/transonic/backends/__init__.py
+++ b/src/transonic/backends/__init__.py
@@ -15,6 +15,7 @@
    cython
    numba
    py
+   jax
 
 User API
 --------
diff --git a/src/transonic/backends/jax.py b/src/transonic/backends/jax.py
index 5827e79..fb96549 100644
--- a/src/transonic/backends/jax.py
+++ b/src/transonic/backends/jax.py
@@ -64,7 +64,11 @@ def make_backend_source(self, info_analysis, func, path_backend):
 
 
 class JaxBackend(PythonBackend):
-    """Main class for the Jax backend"""
+    """Main class for the Jax backend
+
+    See https://github.com/google/jax
+
+    """
 
     backend_name = "jax"
     _SubBackendJIT = SubBackendJITJax