Lightning-AI / Lightning-AI/lightning-thunder

[feature request] Numpy 2 compatibility

Open
#618 1 comment 0 reactions 0 assignees View on GitHub
numpy 2
Dominant language
Python
Stars
1.5k
Forks
121
PR merge metrics
No merged PRs in 30d

Description

It seems like it is quite easy to make Thunder support NumPy 2, modulo dependencies (`fdm`, for example).

A temporary fix could be disabling envs with NumPy 2 in our CIs, https://github.com/Lightning-AI/lightning-thunder/pull/617.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reviewing the CI configuration and dependency declarations, then inspect the fdm dependency and the temporary workaround discussed in pull request #617. Done means NumPy 2 environments are supported without disabling them, with the affected dependencies working correctly.

Written by the indexing model from the issue text.

Assessment

Tech stack
numpy, python
Domain
build-system, ci-cd
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.