[Feature request] prospects of integrating jax's jacfwd, jacrev, and hessian functions in mlx
Open
Nobody has claimed this yet.
enhancement
low priority
- Dominant language
- C++
- Stars
- 28.5k
- Forks
- 2.3k
- Avg merge
- 3d 8h
- Merged PRs (30d)
- 62
Description
Discussed in https://github.com/ml-explore/mlx/discussions/154
Originally posted by LiPingYen December 14, 2023
In JAX, there are jacfwd, jacrev, and hessian functions for transforming the objective function into functions that compute first-order or second-order derivatives. I'm curious to know if MLX has plans to incorporate these three functions in the future.
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
Start by reading the linked Discussion #154 and the JAX functions named in the request: jacfwd, jacrev, and hessian. The issue does not identify MLX files, entry points, tests, or a defined acceptance criterion, so the first step is to clarify the intended API and scope before implementation can be considered done.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100