DEV Community

#jax

Posts

đź‘‹ Sign in for the ability to sort posts by relevant, latest, or top.
I Rewrote My RL Agent in DeepMind's Neural Network Library - Day 9 (Haiku)
madhumithakolkar profile

I Rewrote My RL Agent in DeepMind's Neural Network Library - Day 9 (Haiku)

#jax #reinforcementlearning #machinelearning #deeplearning
5
Comments 1
3 min read
Large-Scale TensorCircuit Contractions on GPUs: Disabling XLA GPU Autotuning
refractionray profile

Large-Scale TensorCircuit Contractions on GPUs: Disabling XLA GPU Autotuning

#jax
Comments
4 min read
One TPU Chip, Eight Agents: Serving Small Agent Workloads with Raw JAX

Goodput vs throughput trade-offs in agent loops
xbill profile

One TPU Chip, Eight Agents: Serving Small Agent Workloads with Raw JAX

#tpu #llm #jax #agents
11
Comments 4
15 min read
PyTrees Are Not One Thing: JAX, PyTorch, and TensorFlow Compared
refractionray profile

PyTrees Are Not One Thing: JAX, PyTorch, and TensorFlow Compared

#jax #torch #tensorflow
Comments
8 min read
Why JAX Is a Much Better Backend for Quantum Circuit Simulation Than PyTorch
refractionray profile

Why JAX Is a Much Better Backend for Quantum Circuit Simulation Than PyTorch

#jax #pytorch
Comments
3 min read
You Can Now Train Quantum Signal Processing Phase Angles with Gradient Descent
blade_dancer_c3 profile

You Can Now Train Quantum Signal Processing Phase Angles with Gradient Descent

#quantum #qml #jax #qsp
Comments 1
3 min read
đź‘‹ Sign in for the ability to sort posts by relevant, latest, or top.