Owner

rdyro

登录后关注

已索引 1 个仓库 · 在 GitHub 查看

torch2jax
Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.

Python · 61 个 star

适合新手全部

未关闭全部

可能已有人在做的 issue 排在最后。全部按时间排序

Torch2Jax vs Torchax?
未关闭
- 哪里不对？
难度 5/5 一周以上新手友好度 20/100

rdyro/torch2jax#32 · 1 条评论 · 2025年11月26日创建
jax 0.7.0 Breaks Torch2Jax
未关闭
- 哪里不对？
难度 4/5 3-5 天新手友好度 45/100

rdyro/torch2jax#31 · 4 条评论 · 2025年7月22日创建
trace_graph fundamentally broken
未关闭
- 哪里不对？
难度 3/5 1-2 天新手友好度 35/100

rdyro/torch2jax#30 · 1 条评论 · 2025年6月28日创建
Jax 0.6.0 now requires cudnn 9.8
未关闭
- 哪里不对？
难度 3/5 1-2 天新手友好度 35/100

rdyro/torch2jax#29 · 3 条评论 · 2025年4月24日创建
[Question] jax.vmap & nested jax.vmap behavior
未关闭
- 哪里不对？
难度 5/5 一周以上新手友好度 25/100

rdyro/torch2jax#28 · 5 条评论 · 2025年3月24日创建
pmap performance
未关闭
- 哪里不对？
难度 4/5 3-5 天新手友好度 35/100

rdyro/torch2jax#25 · 6 条评论 · 1 个 reaction · 2025年1月17日创建
Multi-GPU Question
未关闭
- 哪里不对？
难度 4/5 3-5 天新手友好度 25/100

rdyro/torch2jax#18 · 9 条评论 · 1 个 reaction · 2024年10月27日创建
JAX / XLA adding the importing external dlpack-aware Python arrays.
未关闭
- 哪里不对？
难度 5/5 一周以上新手友好度 15/100

rdyro/torch2jax#6 · 1 条评论 · 2023年8月23日创建

精选适合新手参与的 GitHub issue 摘要。