-
First, sorry to put it in issues, but there is no discussion tab. Second, I don't want this to sound negative or critical, rather I am genuinely interested in the reasoning (and it looks like your lib…
-
For domain randomization it is not particularly easy to `vmap` over different `System` values. For example the `gravity` values, or the `elasticity`. Preferably you should be able to do this in `env.r…
-
I'm running into an issue using the ReversibleHeun solver, which may or may not just be an issue of choosing a proper step size. I've tried to make a MWE that still has the essence of my use case.
…
-
Hi,
Im doing the installation without any problem on a conda environment, but when I try to run the multiomics integration example I receive the error "name 'FlashMHA' is not defined", also, at the b…
-
Here the roadmap for HMC implementation:
- [ ] 0: basic GP regression with model and NUTS from [`pymc3`](https://docs.pymc.io/)
-
There are a bunch of tricks that one needs to know to use PyType with JAX (esp. also in combination with Flax). For example, a PyTree needs to be treated as "Any".
Since it's very common to want t…
-
@jedbrown and I have been discussing the possibility of using JAX to write qfunctions, since it supports JIT compilation and automatic differentiation. I see several ways to go about this, and several…
-
### Report
Hi, when I am running the code in "perturbation_space.ipynb" and I get the to section on logistic regression classifier space:
ps = pt.tl.LRClassifierSpace()
psadata = ps.compute(adata…
-
It would be great if [hyp2f1](https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.hyp2f1.html#scipy.special.hyp2f1) could be added under jax.scipy.special! Are there any plans to do thi…
-
Hello,
Thank you for the great tool. I am currently having difficulty installing cell2location. Below is the method I use to install the software.
```bash
export PYTHONNOUSERSITE="literallyanyl…