repositories Search Results · repo:jacobthebanana/jax-pjit-examples 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

 (136 ms)

0 files

injacobthebanana/jax-pjit-examples (press backspace or delete to remove)

Sort by: Best match

Filter

jacobthebanana/jax-pjit-examples

Leverage JAX pjit to run models that won't fit on a single accelerator.

• Python
·
• 0
·
• Updated
  on Oct 3, 2022

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.