repositories Search Results · repo:rdyro/torch2jax language:Python

Filter by

• • Code... (...)results
  • Issues

    results
  • Pull requests

    results
  • Discussions

    results
  • Commits

    results
  • Packages

    results
  • Wikis

    results

• Advanced

  • ‎Owner‎
  • ‎Size‎
  • ‎Number of followers‎
  • ‎Number of forks‎
  • ‎Number of stars‎
  • ‎Date created‎
  • ‎Date pushed‎
  • ‎Topic‎
  • ‎License‎
  • ‎Archived‎
  • ‎Public‎
  • ‎Private‎
• Advanced search

Code... (...)

0 files

 (143 ms)

0 files

inrdyro/torch2jax (press backspace or delete to remove)

Sort by: Best match

Filter

rdyro/torch2jax

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

• Python
·
• 42
·
• Updated
  25 days ago

Star

Sponsor open source projects you depend on

Contributors are working behind the scenes to make open source better for everyone—give them the help and recognition they deserve.Explore sponsorable projects

ProTip! 

Press the

/

key to activate the search input again and adjust your query.

Sponsor open source projects you depend on

Contributors are working behind the scenes to make open source better for everyone—give them the help and recognition they deserve.Explore sponsorable projects

ProTip! 

Press the

/

key to activate the search input again and adjust your query.