google-deepmind / google-deepmind/alphageometry
TypeError is raised when running alphageometry
- Dominant language
- Python
- Stars
- 4.9k
- Forks
- 572
- PR merge metrics
- No merged PRs in 30d
Description
DDAR can be executed successfully. However, when attempting to test `alphageometry`, I encountered a `TypeError` as follows:
```
TypeError: JAX only supports number and bool dtypes, got dtype bfloat16 in astype
```
I have attempted to identify the bug but have been unsuccessful so far. Could anyone provide assistance? Any help would be greatly appreciated!
Contributor guide
Research direction
Start by reproducing the reported failure when running alphageometry and capture the full traceback, then inspect the code path that reaches JAX astype with bfloat16. The issue is done when alphageometry runs without this TypeError while DDAR continues to execute successfully.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100