02/04: gnu: Add python-jaxtyping.

[guix-commits]

monego pushed a commit to branch master
in repository guix.

commit 70b8682eaa94b5a6517a8fe733eb599ca2a778e3
Author: Vinicius Monego <monego@posteo.net>
AuthorDate: Sat Jul 29 14:52:53 2023 -0300

    gnu: Add python-jaxtyping.
    
    * gnu/packages/machine-learning.scm (python-jaxtyping): New variable.
---
 gnu/packages/machine-learning.scm | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/gnu/packages/machine-learning.scm 
b/gnu/packages/machine-learning.scm
index fd0be8d500..92a60d1616 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -2060,6 +2060,31 @@ physics-informed learning.  It includes implementations 
for the PINN
 MFNN (multifidelity neural network) algorithms.")
     (license license:lgpl2.1+)))
 
+(define-public python-jaxtyping
+  (package
+    (name "python-jaxtyping")
+    (version "0.2.21")
+    (source (origin
+              (method url-fetch)
+              (uri (pypi-uri "jaxtyping" version))
+              (sha256
+               (base32
+                "19qmsnbn4wv2jl99lpn622qs49mrfxmx8s9pr5y8izzgdjq1fvii"))))
+    (build-system pyproject-build-system)
+    ;; Tests require JAX, but JAX can't be packaged because it uses the Bazel
+    ;; build system.
+    (arguments (list #:tests? #f))
+    (native-inputs (list python-hatchling))
+    (propagated-inputs (list python-numpy python-typeguard
+                             python-typing-extensions))
+    (home-page "https://github.com/google/jaxtyping";)
+    (synopsis
+     "Type annotations and runtime checking for JAX arrays and others")
+    (description "@code{jaxtyping} provides type annotations and runtime
+checking for shape and dtype of JAX arrays, PyTorch, NumPy, TensorFlow, and
+PyTrees.")
+    (license license:expat)))
+
 ;; There have been no proper releases yet.
 (define-public kaldi
   (let ((commit "be22248e3a166d9ec52c78dac945f471e7c3a8aa")