patrick-kidger / patrick-kidger/diffrax
FYI: JAX 0.7.0 and 0.7.1 are known to be buggy with Diffrax.
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 2.1k
- Forks
- 189
- Avg merge
- 3d 18h
- Merged PRs (30d)
- 1
Description
Pinning an issue to highlight that when used with Diffrax:
- JAX 0.7.0 is known to cause segfaults (https://github.com/jax-ml/jax/issues/30517)
- JAX 0.7.1 is known to cause major performance issues (https://github.com/jax-ml/jax/issues/31284).
Right now JAX 0.6.* is recommended instead.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
No repository file, test, or entry point is named. Review the issue text and linked JAX reports first; the work would be complete when the project's guidance clearly records the affected JAX versions and the recommended compatible range.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Documentation
- Difficulty
- 1/5
- Estimated time
- Under an hour
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100