cd/entity/JAX· home entities JAX
grep -l @jax /news/*.json | wc -l → 81

JAX

mentions 81 type Organization page 2/5 feed RSS

// recent coverage 81 mentions

16:46
2026-08-16
frontierroles.com
artificial-intelligence

Principal AI Systems Performance Engineer — SambaNova Systems

SambaNova Systems is hiring a Principal AI Systems Performance Engineer in San Jose, California, to optimize and scale state-of-the-art foundation models on its reconfigurable dataflow platform. The r…

10:41
2026-08-15
frontierroles.com
machine-learning

Product ML Engineer — Photoroom

Photoroom, a Y Combinator 2020 cohort company founded in 2019, is hiring a Product ML Engineer in Paris with a salary of €10k/yr, offering a hybrid work model. The role involves owning ML projects end…

19:00
2026-08-14
gilesthomas.com
artificial-intelligence

Why do OpenAI's GPT-2 weights beat mine? Part two: IFT dropout

OpenAI's GPT-2 medium weights achieved the highest IFT score of 42.43 with only 2 IFT epochs, while a JAX model with no dropout and no MHA bias scored 21.45 with 5 epochs, and a JAX model with dropout…

16:08
2026-08-13
developers.googleblog.com
artificial-intelligence

HeyGen x Google Cloud: Bringing Avatar IV to TPUs

HeyGen, an AI video generation platform, announced that its Avatar IV diffusion model, which runs on more than 18B parameters, is now 1.86× faster on Google Cloud's eight-chip Trillium (v6e) TPU host …

18:19
2026-08-12
frontierroles.com
robotics

Senior Reinforcement Learning Engineer — Apptronik

Apptronik, a human-centered robotics company developing AI-powered robots, is hiring a Senior Reinforcement Learning Engineer for its Sunnyvale, CA office to implement and deploy state-of-the-art RL a…

14:52
2026-08-01
frontierroles.com
artificial-intelligence

AI Engineer, Intern — Postman

Postman, the API platform used by over 45 million developers and 500,000 organizations, is hiring an AI Engineer Intern in Berkeley, California, to work on large-scale AI and agentic systems from data…

15:34
2026-07-29
iza.ac
machine-learning

We accidentally built an LLVM compiler for Jax

Xanadu's quantum compiler Catalyst, built for the PennyLane library, accidentally became an LLVM compiler for JAX when users fed it pure JAX NumPy code without quantum instructions, bypassing XLA to l…

13:00
2026-07-29
hiraditya.github.io
machine-learning

When XLA Isn't Enough: Pallas, Mosaic, and Triton

JAX's Pallas kernel system, which lowers through Triton on GPU and Mosaic on TPU, lets developers write custom kernels when XLA's automatic fusion is insufficient for operations like flash attention, …

19:00
2026-07-28
hiraditya.github.io
machine-learning

What JAX Traces, and What It Refuses

JAX, a numerical computing library, is fundamentally a tracing machine that converts pure Python functions into a typed intermediate representation called a jaxpr, refusing to execute code with contro…

19:00
2026-07-25
hiraditya.github.io
machine-learning

XLA Up Close: What It Optimizes, and What It Won't

XLA, the compiler behind JAX, TensorFlow, and PyTorch/XLA, optimizes array programs by freezing shapes, statically allocating buffers, and fusing operations against a global cost model, which makes it…

03:00
2026-07-24
hiraditya.github.io
machine-learning

A Tour of XLA: Where MLIR Lives (and Where It Doesn't)

XLA, the compiler under JAX, TensorFlow, and PyTorch/XLA, uses two intermediate representations: classic HLO (a hand-built C++ IR) and MLIR dialects such as StableHLO and CHLO, with a translation laye…

22:02
2026-07-23
promptcube3.com
artificial-intelligence

DQN vs Q-Learning: Scaling RL with Neural Networks

Deep Q-Networks (DQN) replace the Q-table in Q-learning with a neural network that outputs Q-values for all actions, enabling generalization across similar states. To stabilize training, DQN uses expe…

← prev page 2 / 5 next →
// co-occurs with top 8 entities
// topics top 6 topics