google-deepmind / google-deepmind/alphageometry

TypeError is raised when running alphageometry

Open
#108 1 comment 0 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.