Rare find

Jax2Torch. Run Jax code inside Pytorch projects.

github.com/lucidrains/jax2torch
by @lucidrains

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

July 2023
  • Merge pull request #3 from euclaise/main
  • Capitalization
  • Make it look nicer
  • Update README.md
October 2021
  • port over comment
  • cleanup
  • readme
  • Release —0.0.7
  • support multiple inputs that require grad
  • Release —0.0.6
  • take care of functions that output multiple tensors (say auxiliary lo…
  • Release —0.0.5
  • bump for release
  • make sure one can still pass in keyword arguments in any order, like …
  • Release —0.0.4
  • import torch and jax again
  • Release —0.0.3
  • Release —0.0.2
  • make sure dlpack inits correctly
  • update readme
  • Release —0.0.1
  • github actions
  • readme
  • first commit
  • Initial commit