技术栈
jax
deephub
21 天前
人工智能
·
深度学习
·
google
·
微调
·
大语言模型
·
jax
Google开源Tunix：JAX生态的LLM微调方案来了
JAX生态这两年在LLM训练这块追赶得挺快。PyTorch虽然还是主流但JAX在并行计算、TPU加速和API组合性上确实有些独特的优势。Google今天放出了Tunix这个库，专门做LLM的后训练——微调、强化学习、知识蒸馏这些都能搞。
坚持学习的你
1 年前
人工智能
·
pytorch
·
python
·
jax
Jax(Random、Numpy)常用函数
目录JaxvmapArrayreshapeRandomPRNGKeyuniformnormalsplit
yangxue_mifen
1 年前
人工智能
·
深度学习
·
机器学习
·
jax
recursion depth exceeded” error
有些时候不可以用@jax.jit装饰器参考资料：使用 JAX 后端在 Keras 3 中训练 GAN |由 Khawaja Abaid |中等 (medium.com)
zy_destiny
2 年前
人工智能
·
深度学习
·
机器学习
·
开源
·
jax
·
深度学习框架
·
马斯克
【马斯克开源GROK-1模型】意味着什么？
目录1.激动人心的消息Grok-1 根据 Apache License 2.0 开放源代码题外话-介绍JAX框架
我是有底线的