diff --git a/src/basic-languages/swift/swift.ts b/src/basic-languages/swift/swift.ts
index c7adefdc..a5f7a41e 100644
--- a/src/basic-languages/swift/swift.ts
+++ b/src/basic-languages/swift/swift.ts
@@ -57,6 +57,8 @@ export const language = <languages.IMonarchLanguage>{
 		'@asyncHandler',
 		'@available',
 		'@convention',
+		'@derivative', // Swift for TensorFlow
+		'@differentiable', // Swift for TensorFlow
 		'@discardableResult',
 		'@dynamicCallable',
 		'@dynamicMemberLookup',