-
Hi,
I'm working on a research problem that involves computations of Bessel functions of the first kind, of multiple different integer orders, and I'm trying to leverage JAX for autodiff and fast co…
-
There is no discussion tab for Haiku so I am raising this question here. My apologies in advance if this isn't appropriate. Suggestions of where I might ask would be much appreciated.
The mnist exa…
-
I need this for finding the eigenvectors of the Hessian after tridiagonalizing it with Lanczos iteration. Right now the function looks like:
```python
def eigh_tridiagonal(d: ArrayLike, e: ArrayLi…
-
Hello, I have been using Jax to build a differentiable fluid simulator, which involves solving a sparse linear system using the Conjugate Gradient algorithm (`jax.scipy.sparse.linalg.cg`). When I trie…
-
Here's an overview of the features we intend to work on in the near future.
## Core Keras
### Saving & export
- Implement saving support for sharded models (sharded weights files).
- Improve…
-
Virtual Machine: AWS EC2 g5.2xlarge, NVIDIA A10G GPU, 24G GPU memory
When running vit_jax.ipynb, get_accuracy(params_repl) in the Evaluation section, I got the following error:
```
INFO:absl:Load d…
-
### Description
I've been reading through the autodidax documentation, which is awesome! Just after the evaluation interpreter named `EvalTrace` is introduced, there is a code snippet that shows how …
-
## Issue description
Trying computing the model jacobian using a keras model returns zeros
## Code example
```python
import keras
import torch
model = keras.Sequential(
[keras.layers.De…
-
### Report
Hello,
When running the lda visualisation plot as part of the mixscape workflow, I am not getting consistent results between runs, the plot looks slightly different (although very simil…
-
Currently the only RESt framework supported by the extension is RestEasy, but if for example you are using Apache TomEE you would like to run your client tests with Apache CXF (so no more dependencies…