Nov. 14, 2023, 2:02 a.m. | /u/Llamas1115

Machine Learning www.reddit.com

Have any of you seen benchmarks comparing the performance of \`@jax.jit\` and \`@torch.compile\`, especially when using functional PyTorch code? Are the performance differences big? Small? Do they depend a lot on what you're doing?

benchmarks big code differences functional jax jit machinelearning performance pytorch small torch

Lead Developer (AI)

@ Cere Network | San Francisco, US

Research Engineer

@ Allora Labs | Remote

Ecosystem Manager

@ Allora Labs | Remote

Founding AI Engineer, Agents

@ Occam AI | New York

AI Engineer Intern, Agents

@ Occam AI | US

AI Research Scientist

@ Vara | Berlin, Germany and Remote