”jax“ 的搜索结果

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

Beautiful and accessible math in all browsers

翻译 - 所有浏览器中的数学运算都很漂亮

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

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

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

Model parallel transformers in JAX and Haiku

翻译 - 在 JAX 和 Haiku 中模拟并行转换器

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

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

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

Differentiable, Hardware Accelerated, Molecular Dynamics

Multi-Agent Reinforcement Learning with JAX

SBX: Stable Baselines Jax (SB3 + Jax)

Gaussian processes in JAX.

Monte Carlo tree search in JAX

Computational Fluid Dynamics in JAX

RL Environments in JAX 🌍

Samples about JAX-RS

100 exercises to learn JAX

翻译 - 学习 JAX 的 100 个练习

minGPT in JAX

State Space Models library in JAX

Additive manufacturing simulation with JAX.

A Graph Neural Network Library in Jax

Trainable AlphaFold implementation in JAX

#大语言模型#A simple, performant and scalable Jax LLM!

Jaxwell is JAX + Maxwell

Rigid transforms + Lie groups in JAX