Proprietário · jax-ml

jax-ml

Repositórios de código aberto acompanhados de jax-ml, ordenados por estrelas.

2 repositórios

#1
jax ↗@jax-mlPython
Transformações componíveis de programas Python+NumPy: diferenciação, vetorização, JIT para GPU/TPU e mais
★ 36.241+16Variação de estrelas nos últimos 7 dias
#2
scaling-book ↗@jax-mlHTML
Repositório de "How To Scale Your Model", um livro didático em estilo de blog sobre o escalonamento de LLMs em TPUs
★ 1.396+12Variação de estrelas nos últimos 7 dias