-
### Description
Jax' transfer guard potentially has two problems related to the CPU-placed arrays.
1. Transfering numpy (or pure python) arrays/scalars to XLA-CPU device is considered to be host2…
-
Initially we had only single impl based on RestEasy but now we've got Jersey and pure JAX-RS implementations, so this annotation may be confusing.
-
### Description
Hi! I have a model that requires precomputing a large matrix that is a model constant. To avoid having to compute it at every function call, I thought I'd make it evaluate at compile …
-
When profiling FP, runtime seems to be dominated by `MemcpyD2H` as seen in the Perfetto UI. The actual computations take only very little time in between.
![image](https://user-images.githubusercon…
-
### Description
I am developing an FEM solver using JAX using sparse matrices. The code works perfectly fine on cpu but the exact same code fails on gpu by giving this error:
jaxlib.xla_extensio…
-
i am running alphafold v2.3.2 to predict a multimer on 16CPU,241G,4* V100
```
>3JA9_1|Chains A|Proliferating cell nuclear antigen|Homo sapiens (9606)
MFEARLVQGSILKKVLEALKDLINEACWDISSSGVNLQSMDSSHV…
-
With JAX-RS 2.0 initial support for HATEOAS was introduced with building Link headers (server side) and following them (client side).
With JAX-RS 2.1 this support should get improved by a declarative…
-
Why not add support for [attr classes](https://www.attrs.org/en/stable/index.html)? They're basically namedtuples under steroids.
Looks to me like it would need a small contrib to the pytree file, …
-
### Review Mojo's priorities
- [X] I have read the [roadmap and priorities](https://docs.modular.com/mojo/roadmap.html#overall-priorities) and I believe this request falls within the priorities.
###…
-
## Summary
Requesting the [betaincinv](https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.betaincinv.html) function be added to `jax.scipy.special`. This is the inverse of the regular…