jax

英文: Jax: generating random numbers under **JIT** 问题 Here's the translated code portion: 我有一个设置，需要生成一...

英文: Installing jaxlib for cuda 11.8 问题 我正在尝试在我的Ubuntu 18上安装jax和jaxlib，使用Python 3.8，为了snerg（https://g...

英文: XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed 问题 2023-07-31 01:53:45.0...

英文: pytorch and jax networks give different accuracy with same settings 问题 The issue in the JAX code...

英文: How to make a function a valid jax type? 问题 当我将使用以下函数创建的对象传递到jax.lax.scan函数中时： def logdensity_cr...

英文: How to slice jax arrays using jax tracer? 问题 我正在尝试修改一个代码库，使用现有数组和Jax跟踪器形式的索引来创建子数组。当我尝试直接将这些Jax跟...

英文: Creating a jax array using existing jax arrays of different lengths throws error 问题 我正在使用以下代码来使用...

英文: Error using JAX, Array slice indices must have static start/stop/step 问题 I'll be happy to help y...

英文: How can I test if a jitted Jax function creates new tensor or a view? 问题 I have a basic code lik...

英文: Understanding how JAX's tracer vs static work 问题 以下是您提供的代码的翻译： 我是JAX的新手，试图使用JAX编写一个简单的代码，在某个...