search

LouisDesdoigts / zodiax

Object-oriented Jax framework extending Equinox for scientific programming
https://louisdesdoigts.github.io/zodiax/
BSD 3-Clause "New" or "Revised" License
11 stars 1 forks source link

Gradient of float returns None #35

Open maxecharles opened 10 months ago

maxecharles commented 10 months ago

Doing a gradient descent, and the gradient of a parameter of type float always returns None. This was fixed by changing it to a jax array.