issues Search Results · repo:microsoft/folx language:Python
Filter by
13 results
(130 ms)13 results
inmicrosoft/folx (press backspace or delete to remove)Description
Hello, I encountered a TypeError when computing the gradient of a function inside jax.shard_map including
jnp.linalg.slogdet.
The specific error is: File /users/ylz/.conda/envs/jax_0_7_1/lib/python3.13/site-packages/jax/_src/interpreters/ad.py ...
zhangylch
- Opened on Dec 15, 2025
- #40
Recently came across this somewhat obscure error because of a missing pvary. The missing pvary is the one I mentioned at
the end of https://github.com/microsoft/folx/pull/36#issue-3388257713, but there ...
inailuig
- Opened on Nov 22, 2025
- #38
Thank you for implementing the forward Lap. It significantly accelerates the Laplacian calculation. I encountered a
TypeError when computing the Laplacian of a function involving jnp.linalg.slogdet using ...
- 2
- Opened on Nov 21, 2025
- #37
Hello, thank you for the great library — it makes the Laplacian calculations much faster! However, I m having trouble
with sparsity when trying to use folx with a LapNet-like architecture that includes ...
mmezic
- Opened on Jun 4, 2025
- #35
In some recent jax version, the pallas API changed from
pl.pallas_call(
...,
compiler_params=dict(triton=dict(
num_warps=num_warps,
num_stages=num_stages,
)),
...
)
to ...
ae-foster
- Opened on May 27, 2025
- #33
Hi,
I m trying to use folx (0.2.15) with the latest JAX (0.6.0), but get an import error when I try
import folx
that looks as follows:
[...]
from jax.core import Primitive
ImportError: cannot import ...
f-dangel
- 3
- Opened on Apr 26, 2025
- #30
The mask in jvp might not match the pytree structure of grad_y.
Unfortunately, I don t have an open-source repro at the moment, but these lines assume pytrees of a similar structure,
however the mask ...
rdyro
- 2
- Opened on Apr 11, 2025
- #29
Hi
Thank you for implementing the forward Lap. It significantly accelerates the Laplacian calculation. However, I am
encountering an error when attempting to use sparsity. It seems that the sparsity is ...
- 4
- Opened on Nov 17, 2024
- #28
Hi,
I have encountered the following warnings of unregistered functions. These functions add_any and pad seem to be JAX
primitives and I don t find any intructions in README.md for how to register them. ...
zgbkdlm
- 1
- Opened on Nov 4, 2024
- #27
@n-gao Is bumping to python 3.11 absolutely necessary?
We are using ferminet repo on rocm which has folx dependency. However, we can t force python 3.11 (At least not
immediately). I am sure there are ...
Ruturaj4
- 5
- Opened on Sep 5, 2024
- #25
Learn how you can use GitHub Issues to plan and track your work.
Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub IssuesProTip!
Restrict your search to the title by using the in:title qualifier.Learn how you can use GitHub Issues to plan and track your work.
Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub IssuesProTip!
Restrict your search to the title by using the in:title qualifier.