Python jax_专题 - Python中文网

当前热门话题： Python jax: 本站为您提供最新、最全的jax的Python教程、文档、代码、资源等相关内容，Python中文网技术交流社区同时还提供学习资源下载，如：电子书、IDE编辑器软件、编程视频、代码规范标准、WEB开发、GUI、科学计算与人工智能等相关内容。

本文网址：https://www.cnpython.com/tags/289216
欢迎加入QQ群-->： 979659372

关于jax 相关联的Python项目和问题：

在gunicorn/flask服务器中使用google的JAX

我想用flask和gunicorn为Google JAX框架内处理数据的应用程序提供服务如果在烧瓶内运行，一切正常。只要我在gunicorn中运行应用程序，每个与jax相关的部分都会导致工作进程死 ...

2024-05-24 已阅读: n次

如何在编译的jax代码中执行非整数索引算法？

如果我们对数组索引执行非整数计算（然后转换为int（）），那么在jit编译的jax代码中似乎仍然无法将结果用作有效索引。我们如何解决这个问题下面是一个简单的例子。具体问题：命令jnp.diag_in ...

2024-05-24 已阅读: n次

如何为softmax编写JAX自定义矢量雅可比产品（vjp）

为了理解JAX的反向模式auto diff，我尝试为softmax编写一个自定义的_vjp，如下所示： import jax import jax.numpy as jnp import numpy ...

2024-05-24 已阅读: n次

Django和Ajax。无限卷轴实现的正确方法

我有一个页面，用户在其中选择一些标记，然后在我的视图中，我对mongodb进行聚合查询，并根据所选的标记返回结果。有很多文档，这就是为什么我只显示固定数量的文档。我想在ajax中使用无限滚动。在所以 ...

2024-05-24 已阅读: n次

BF16和多个GPU的JAX/FLAX运行时错误

我正在编写最新版本的T5X（亚麻代码）。虽然在Colab（带有一个GPU的标准笔记本电脑）上运行它没有问题，但在我的一台机器上运行代码时，我遇到了一些困难。具体地说，我的配置是Ubuntu 18.0 ...

2024-05-24 已阅读: n次

从头开始实现二元交叉熵会导致神经网络的训练结果不一致

我正在尝试使用JAX库及其小的神经网络子模块“Stax”来实现和训练一个神经网络。因为这个库没有二进制交叉熵的实现，所以我写了自己的： def binary_cross_entropy(y_hat, ...

2024-05-24 已阅读: n次

乙状结肠的Jax自动标记总是返回nan

我试图区分一个函数，该函数近似于一个高斯分数，该分数包含在给定移动平均值的2个极限（一个截断高斯）内。 jnp.grad不允许我区分相加布尔过滤器（注释行），因此我不得不临时使用一个sigmoid 然 ...

2024-05-24 已阅读: n次

Jaxlib pip安装失败

在命令行中，我尝试了以下安装tutorial如果可能的话，我希望避免从源代码构建。目前，我不确定问题是什么。是否有人可以验证在尝试安装Jaxlib时是否得到相同/不同的响应为了提高认识，Jax安装得 ...

2024-05-24 已阅读: n次

使用numpy和jax进行非Transitive子类化

我的问题很简单： >>> isinstance(x, jax.numpy.ndarray) True >>> issubclass(jax.numpy.ndarra ...

2024-05-24 已阅读: n次

Jax找不到静态argnums

这与this question有关。除了一件奇怪的事情外，我设法使大部分代码都能正常工作这是修改后的代码 import jax.numpy as jnp from jax import grad, ...

2024-05-24 已阅读: n次

如何将特殊格式的文本组织到列表中（在python上）

我有航班代码的数据。它有很多这样的字符串： routing = 'PBI-FLL/FMY/JAX/MIA/ORL-PNS/TPA-SRQ-CLE/CHI/HOU/WAS-DEN-ELP' 我需要 ...

2024-05-24 已阅读: n次

如何在JAX中创建非完全连接的卷积层？

我试图在JAX中实现LeNet5。我想我可以完成大部分工作，以下是架构供参考：我想我知道如何实现前两层： c1 = stax.Conv(out_chan=6, filter_shape=(5, 5 ...

2024-05-24 已阅读: n次

fastax

快斧概述 Fastax是一个深度学习库，建立在Google的JAX之上，它试图通过将jax的多功能性与最常用的神经网络层和实用程序。更多具体信息和命名 fastax是fastai和jax、fa ...

2024-05-24 已阅读: n次

pantsbuild.pants.contrib.jax-ws

1.19.x稳定释放本文档介绍了 1.19.x 稳定系列之前的版本。 1.19.0（2019年8月21日） ...

2024-05-24 已阅读: n次

jaxnet

jaxnet JAXNET是一个基于^ {A3}的深度学习库。 jaxnet的函数式api比tensorflow2、keras和pytorch提供了独特的优势，同时保持用户友好性、模块性和可扩展性 ...

2024-05-24 已阅读: n次

jax

区分、编译和转换numpy代码。此包Python名称：jax 目前版本： jax 0.1.43 最后维护时间：Aug 25, 20 ...

2024-05-24 已阅读: n次

jaxlib

用于jax的xla库此包Python名称：jaxlib 目前版本： jaxlib 0.1.27 最后维护时间：Aug 27, 20 ...

2024-05-24 已阅读: n次