Swiftpack.co - owkin/GrAIdient as Swift Package

Swiftpack.co is a collection of thousands of indexed Swift packages. Search packages.
See all packages published by owkin.
owkin/GrAIdient v0.2.0
GrAIdient is a deep learning framework that aims at challenging the way we train and run models in computer vision.
⭐️ 56
🕓 3 weeks ago
.package(url: "https://github.com/owkin/GrAIdient.git", from: "v0.2.0")

Ready for the grAIt descent?

GrAIdient is a framework that allows deep learning models to be developed using the internal GPU of a Mac, unlocking researchers to more easily train and run AI models on their own computers.

GrAIdient exposes the graph of layers, providing a unique way to design deep learning models for greater understanding, control and reproducibility.

Though deeply grounded to the data driven pipeline, the goal is to challenge the very understanding of deep learning models and inject human intelligence where relevant; to transition from black box models to white box models, and all the gradients in between.

Check out this toy VGG example and its documentation to get started with GrAIdient today!

✨ Key Features

  • direct access to the graph of layers and to the backward pass
  • run natively on Mac GPU (Intel GPU, eGPU, Apple Silicon)
  • written in Swift: a compiled language with strong typing
  • gradient checking to validate backward operations
  • PyTorch interoperability
  • gradients per batch & per sample (needed for differential privacy)
  • debug at the neuron level

📦 Swift Package Manager

Add the following dependency to your Package.swift manifest:

.package(url: "https://github.com/owkin/GrAIdient.git", .branch("main")),

📚 Documentation

The documentation is divided into several sections:

👨‍💻 Contributing

Read below to learn how to take part in improving GrAIdient.


All notable changes to GrAIdient will be documented in the changelog.

Contributing Guide

Read our contributing guide to learn about our development process and how to build and test your changes to GrAIdient.

Code of Conduct

GrAIdient has adopted a Code of Conduct that we expect project participants to adhere to. Please read the full text so that you can understand what actions will and will not be tolerated.


GrAIdient, GrAITestsUtils and GrAITests are MIT licenced.

GrAIExamples and GrAITorchTests both depend on PythonKit and are Apache 2.0 licensed.


Stars: 56
Last commit: 8 hours ago
jonrohan Something's broken? Yell at me @ptrpavlik. Praise and feedback (and money) is also welcome.

Release Notes

4 weeks ago

0.2.0 (2023-02-27)


🪜 layer_1d: Softmax1D, DotProduct1D & Constant1D (#49)
🪜 feat: remove activation from layer (#47)
🪜 feat: LayerMerge1D, Sum1D, Concat1D, Concat2D (#43)
🪜 layer_2d: Deconvolution2D (#42)
🪜 feat: getDeltaWeightsGPU per sample API (#41)

Bug Fixes

🐛 fix: use buffers for neuron selection in SelectNeurons1D (#50)
🐛 fix: model context max id (#45)
🐛 fix: remove error when data input may indicate lower batch size (#44)

Miscellaneous Tasks

📚 docs: change project description and add links (#57)
📚 docs: PropertyListEncoder by default (#51)
🎉 refactor: logo (#46)
🎉 refactor!: re brand the framework (#40)

Swiftpack is being maintained by Petr Pavlik | @ptrpavlik | @swiftpackco | API | Analytics