fbpic
Numba
fbpic | Numba | |
---|---|---|
2 | 124 | |
165 | 9,452 | |
-0.6% | 1.1% | |
8.1 | 9.9 | |
9 days ago | 8 days ago | |
Python | Python | |
GNU General Public License v3.0 or later | BSD 3-clause "New" or "Revised" License |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
fbpic
-
Numba: A High Performance Python Compiler
When I wrote my bachelor thesis years back I worked on a particle-in-cell code [1] that makes heavy use of numba for GPU kernels. At the time it was the most convenient way to do that from python. I remember spending weeks to optimizing these kernels to eek out every last bit of performance I could (which interestingly enough did eventually involve using atomic operations and introducing a lot of variables[2] instead of using arrays everywhere to keep things in registers instead of slower caches).
I remember the team being really responsive to feature requests back then and I had a lot of fun working with it. IIRC compared to using numpy we managed to get speedups of up to 60x for the most critical pieces of code.
[1]: https://github.com/fbpic/fbpic
-
Faster Python calculations with Numba: 2 lines of code, 13× speed-up
We used numba to accelerate the code and most importantly write GPU kernels for the heavy parts. I remember spending hours optimising my code to eek out the most performance possible (which eventually meant using atomics and manually unrolling many loops because somehow this was giving us the best performance) but honestly I was really happy that I didn't need to write cuda kernels in C and generally it was pretty easy to work with. I remember back then the documentation was sometimes a little rough around the edges but the numba team was incredibly helpful and responsive. Overall I had a great time.
[0] https://github.com/fbpic/fbpic
Numba
-
Mojo🔥: Head -to-Head with Python and Numba
Around the same time, I discovered Numba and was fascinated by how easily it could bring huge performance improvements to Python code.
-
Is anyone using PyPy for real work?
Simulations are, at least in my experience, numba’s [0] wheelhouse.
[0]: https://numba.pydata.org/
-
Any data folks coding C++ and Java? If so, why did you leave Python?
That's very cool. Numba introduces just-in-time compilation to Python via decorators and its sole reason for being is to turn everything it can into abstract syntax trees.
- Using Matplotlib with Numba to accelerate code
-
Python Algotrading with Machine Learning
A super-fast backtesting engine built in NumPy and accelerated with Numba.
-
PYTHON vs OCTAVE for Matlab alternative
Regarding speed, I don't agree this is a good argument against Python. For example, it seems no one here has yet mentioned numba, a Python JIT compiler. With a simple decorator you can compile a function to machine code with speeds on par with C. Numba also allows you to easily write cuda kernels for GPU computation. I've never had to drop down to writing C or C++ to write fast and performant Python code that does computationally demanding tasks thanks to numba.
-
Codon: Python Compiler
Just for reference,
* Nuitka[0] "is a Python compiler written in Python. It's fully compatible with Python 2.6, 2.7, 3.4, 3.5, 3.6, 3.7, 3.8, 3.9, 3.10, and 3.11."
* Pypy[1] "is a replacement for CPython" with builtin optimizations such as on the fly JIT compiles.
* Cython[2] "is an optimising static compiler for both the Python programming language and the extended Cython programming language... makes writing C extensions for Python as easy as Python itself."
* Numba[3] "is an open source JIT compiler that translates a subset of Python and NumPy code into fast machine code."
* Pyston[4] "is a performance-optimizing JIT for Python, and is drop-in compatible with ... CPython 3.8.12"
[0] https://github.com/Nuitka/Nuitka
[1] https://www.pypy.org/
[2] https://cython.org/
[3] https://numba.pydata.org/
[4] https://github.com/pyston/pyston
-
This new programming language has the potential to make python (the dominant language for AI) run 35,000X faster.
For the benefit of future readers: https://numba.pydata.org/
-
Two-tier programming language
Taichi (similar to numba) is a python library that allows you to write high speed code within python. So your program consists of slow python that gets interpreted regularly, and fast python (fully type annotated and restricted to a subset of the language) that gets parallellized and jitted for CPU or GPU. And you can mix the two within the same source file.
- Numba Supports Python 3.11
What are some alternatives?
WarpX - WarpX is an advanced, time-based electromagnetic & electrostatic Particle-In-Cell code.
NetworkX - Network Analysis in Python
simsopt - Simons Stellarator Optimizer Code
jax - Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
pure_numba_alias_sampling - Pure numba version of Alias sampling algorithm from L. Devroye's, "Non-Uniform Random Random Variate Generation"
Dask - Parallel computing with task scheduling
autograd - Efficiently computes derivatives of numpy code.
cupy - NumPy & SciPy for GPU
ndarray_comparison - Benchmark of toy calculation on an n-dimensional array using python, numba, cython, pythran and rust
Pyjion - Pyjion - A JIT for Python based upon CoreCLR
rust-numpy - PyO3-based Rust bindings of the NumPy C-API
SymPy - A computer algebra system written in pure Python