Pushing back the limits on numerical computing.
15
Публичные репозитории
39 172
Всего звезд
968
Подписчики
Организация jax-ml на GitHub представляет собой активное сообщество, занимающееся числовыми вычислениями с использованием таких языков, как Python, C++ и HTML. Среди их заметных репозиториев находятся jax, scaling-book и jax-triton, которые охватывают широкий спектр методов и инструментов для разработки и масштабирования моделей машинного обучения.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
jax-triton contains integrations between JAX and OpenAI Triton
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Описание для этого репозитория не предоставлено.
Minimal yet performant LLM examples in pure JAX
State of the art inference for your bayesian models.
Minimal, lightweight JAX implementations of popular models.
Inference Combinators in JAX
Описание для этого репозитория не предоставлено.
Описание для этого репозитория не предоставлено.
Описание для этого репозитория не предоставлено.
Описание для этого репозитория не предоставлено.
Описание для этого репозитория не предоставлено.
jax-ml разрабатывает различные инструменты и библиотеки, включая jax для трансформации Python+NumPy программ, а также scaling-book, который посвящен масштабированию моделей на TPU.
Основными языками программирования, используемыми в репозиториях jax-ml, являются Python, C++, HTML и Jupyter Notebook, что позволяет создавать разнообразные решения в сфере числовых вычислений и машинного обучения.
Да, все репозитории jax-ml являются публичными, что позволяет другим разработчикам и исследователям использовать их инструменты и участвовать в сообществе, делясь своими идеями и улучшениями.
Следите за jax-ml с помощью RepoGuard и получайте уведомления в момент появления нового публичного репозитория.
Следить за этим аккаунтом