Jax - DEV Community

Skip to content

DEV Community

👋 Sign in for the ability to sort posts by relevant, latest, or top.

Madhumitha Kolkar

Jul 21

I Built a Neural Network Training Loop in 5 Lines Using JAX - Day 2

#jax #machinelearning #python #deeplearning

3 min read

Madhumitha Kolkar

Jul 20

I Started Learning JAX as a Senior ML Engineer - Here's My First Impression.

#jax #machinelearning #python #beginners

3 min read

$refractionray profile$

Jul 15

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

#jax

4 min read

$refractionray profile$

Jun 12

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

#jax #torch #tensorflow

8 min read

$refractionray profile$

Jun 6

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

#jax #pytorch

3 min read

Jun 11

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

#quantum #qml #jax #qsp

3 min read

👋 Sign in for the ability to sort posts by relevant, latest, or top.