fix: minor fixes

Author: avik-pal

src/TracedRNumber.jl

@@ -19,6 +19,8 @@ Base.getindex(a::TracedRNumber{T}) where {T} = a
 Base.zero(::TracedRNumber{T}) where {T} = promote_to(TracedRNumber{T}, zero(T))
 Base.one(::TracedRNumber{T}) where {T} = promote_to(TracedRNumber{T}, one(T))
 
+Base.eps(::Type{TracedRNumber{T}}) where {T} = promote_to(TracedRNumber{T}, eps(T))
+
 function Base.convert(::Type{<:TracedRNumber{T}}, x::Number) where {T}
     return promote_to(TracedRNumber{T}, T(x))
 end

src/XLA.jl

@@ -228,7 +228,7 @@ end
 @inline primitive_type(::Type{Float32}) = 11
 
 @static if isdefined(Core, :BFloat16)
-    @inline primitive_type(::Type{BFloat16}) = 16
+    @inline primitive_type(::Type{Core.BFloat16}) = 16
 end
 
 @inline primitive_type(::Type{Float64}) = 12