-
```julia
using CUDA, Enzyme
w = cu(rand(Float32, 10, 10))
b = cu(rand(Float32, 10))
x = cu(rand(Float32, 10, 10))
act = x -> max(0f0, x)
function loss_function(act, w, x, b)
return sum(…
-
Copying over and summarizing some discussion from Slack:
@wsmoses:
> 1. Type-stable deep learning libraries. So far all the DL library code I've seen people try to AD with Enzyme is very very type…
-
- Should we support primitives beyond pushforwards / pullbacks ?
- Requires making assumptions on argument types
- Should we support higher order derivatives?
- Requires mixing backends, ergo …
-
In the Enzyme setups https://github.com/gdalle/DifferentiationInterface.jl/blob/main/DifferentiationInterface/ext/DifferentiationInterfaceEnzymeExt/forward_onearg.jl#L13 it looks like you're using the…
-
If I understand the implementation properly (although the docs are not very clear about the implementation details), this library works by [acquiring the Python GIL](https://github.com/NIFTy-PPL/JAXbi…
-
Hey! I have been playing with this package to implement AD for [WaterLily.jl](https://github.com/WaterLily-jl/WaterLily.jl). The ForwardDiff backend works as expected, and it provides the same result …
b-fg updated
4 months ago
-
MWE:
```julia
using CUDA, Enzyme, Random
rng = MersenneTwister(1234)
m = 32
n = 16
Z = cu(randn(rng, Float32, (n,m)))
𝒯 = 2.0
Δτ = 0.1
ca_init = cu([zeros(1) ; ones(m)])
function f!(ċȧ…
-
Hello, I'm the developer of [mpi4jax](https://github.com/mpi4jax/mpi4jax).
I'd like to thank you for all your effort in wrapping MPI within Python that made our life already much easier when trying …
-
I got an error when use jit.script on some new layers(implemented in c++):
```
RuntimeError:
attribute lookup is not defined on python value of type 'FunctionMeta':
@torch.jit.script_method
def f…
-
Hey,
this project is quite interesting!
However i've noticed that there might be an issue when trying to run the example.
I've tried it with Rust 1.72.0 on a Win11 machine.
Let me know if i can he…