jax

#计算机科学#  Keras是一个基于 Python 的深度学习库，能够在TensorFlow、Microsoft Cognitive Toolkit、Theano或PlaidML之上运行。

  Python+NumPy 程序的可组合转换：自动微分、向量化、JIT即时编译、GPU/TPU 等

#计算机科学#  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

翻译 - 重塑了深度学习操作（支持tf，pytorch，chainer，gluon等）

#计算机科学#  Trax — Deep Learning with Clear Code and Speed

翻译 - Trax –您进阶深度学习的道路

#计算机科学#  Python code for "Probabilistic Machine learning" book by Kevin Murphy

翻译 - 用于“机器学习：概率视角”的Python代码（第2版）

  Flax is a neural network library for JAX that is designed for flexibility.

翻译 - Flax是JAX的神经网络生态系统，旨在提高灵活性。

#计算机科学#  Scenic: A Jax Library for Computer Vision Research and Beyond

翻译 - Scenic：用于计算机视觉研究及其他领域的 Jax 库

#大语言模型#  Training and serving large-scale neural networks with auto parallelization.

#计算机科学#  JAX-based neural network library

翻译 - 基于JAX的神经网络库

#计算机科学#  可重定向的基于 MLIR 的机器学习编译器和运行时toolkit。

#计算机科学#  A library for scientific machine learning and physics-informed learning

#自然语言处理#  Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

  Monte Carlo tree search in JAX

  Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

#计算机科学#  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization i...

#计算机科学#  An Extensible Deep Learning Library

#大语言模型#  Distributed ML Training and Fine-Tuning on Kubernetes

#计算机科学#  Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

翻译 - “大转移（BiT）：通用视觉表示学习”论文的官方资料库。

  Mastering Diverse Domains through World Models

  A JAX research toolkit for building, editing, and visualizing neural networks.