search

salesforce / jaxformer

Minimal library to train LLMs on TPU in JAX with pjit().
BSD 3-Clause "New" or "Revised" License
277 stars 37 forks source link

added deepspeed inference #29

Closed mlap1n closed 1 year ago

salesforce-cla[bot] commented 1 year ago

Thanks for the contribution! Before we can merge this, we need @mlap1n to sign the Salesforce.com Contributor License Agreement.