Comments (4)
It definitely should be possible and I agree the Julia front end is likely a good place to start (also happy to video call next week and do some planning).
My bet as to the big things that will need to be done for Numba as a front end:
- Load Enzyme as a plugin and call it as an optimization at the appropriate time (Julia has good example of)
- Expose API for gradients in numba
- Ensure Numba passes down all of the relevant bitcode
- Ensure Numba passes down Type Information that can be understood (as TBAA)
- Re-entry into Numba JIT if AD of non fully statically analzable code is desired
from enzyme.
You could take a look of how the Julia frontend does things, https://github.com/wsmoses/Enzyme.jl
GPUCompiler works similar to Numba, and there is a C-API call for creating the enzyme pass.
Happy to chat about the detailed strategy, but I am not familiar enough with Numba to make any direct recommendations.
from enzyme.
Hi !
Is there any news on this potential functionality ?
Thanks !
from enzyme.
Hi, would just like to add a +1 that this would be of great interest to me. Have any steps been made or anyone else taken a look in the last couple years?
from enzyme.
Related Issues (20)
- Enzyme crash HOT 3
- Performance of type-analysis HOT 6
- Many failures when running ninja check-enzyme HOT 2
- Enzyme: Cannot cast __enzyme_autodiff primal argument 16 HOT 7
- enzyme_dupped parameter doesn't return gradient
- New C++ interface with lambda HOT 2
- Can't compile eigensumsqdyn-notmp.cpp with Eigen 3.4.0
- Injected headers for c++ break tooling
- Incorrect derivative result when nested void functions and recursive nature functions are used. HOT 6
- abort cmake when -DLLVM_DIR is an invalid path.
- check-enzyme-integration tests failures HOT 3
- Branch mismatcharg fails to compile HOT 6
- Building Enzyme CMake - Undefined symbol: main HOT 3
- Unnecessary caching for recursive functions
- Bug in Enzyme gsl branch HOT 6
- compilation slowdown associated with PreserveNVVMNewPM HOT 1
- Is this N/3 correct? HOT 10
- incorrect derivative of function that returns struct HOT 2
- C++ interface templates appear to be broken HOT 9
- `std::vector.push_back()` causes segementation fault in Enzyme HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from enzyme.