jax

#自然语言处理#为 Jax、PyTorch 和 TensorFlow 打造的先进的自然语言处理

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

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

#计算机科学#🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

#计算机科学#The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

翻译 - 🔥用于可视化和跟踪机器学习实验的工具。此存储库包含CLI和Python API。

#计算机科学#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