-
Hi,
I am currently installing the latest Alphafold (from main branch, so 2.3.2 plus new features, new openmm=7.7.0 , no openmm patch, python=3.10 etc..) on a HPC cluster without Docker. I am con…
-
Not sure if this is a jax thing or dm-haiku... but recently I've been trying to use Conv2DTranspose in my model, and even for very simple case... it takes forever to compile.
here is an example:
`…
-
I find myself often copying setup code (e.g. for a simple MNIST-like example) when trying out new features.
It would be very handy to have specially crafted Snippets for Flax development for this p…
-
Hey all!
I like using mypy as a smell test for my code and I was wondering if Haiku (and to a larger extent, Jax) has (well supported) built in typing? I tried doing
```
import jax.np as jnp
…
-
Hi, I am trying to run it, but the `python3 ./run.py` process exits eventually after it's been running for about 10 minutes at 800% cpu usage.
I am running it in a K8s pod (with /dev/shm of 640Gi; 58…
-
## Expected Behavior
Run Prediction block will complete
## Current Behavior
Run prediction block is currently erroring with AttributeError: module 'jax' has no attribute 'xla'
## Steps to Reproduc…
-
I'm looking for write a basic RNN that does f(Ax+b) at each time step.
What would be the best way to go about it? Could you outline some code to give an idea?
Can one apply JIT over the entire (…
jli05 updated
4 years ago
-
## Paste the link of the GitHub organisation below and submit
https://github.com/deepmind
-
I used this installation environment without any errors throughout the process, and you can also do the same. I am using cuda11 and cudnn8.2, python version 3.8。jax_releases can found here https://sto…
-
Hi,
I have tried multiple combinations of Jax, jaxlib and jax_md versions, but every time, I get the following error.
Please help me out in resolving this issue. Thank you for your time and consider…