Our great sponsors
-
Oceananigans.jl
🌊 Julia software for fast, friendly, flexible, ocean-flavored fluid dynamics on CPUs and GPUs
-
pyhpc-benchmarks
A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python :rocket:
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
like some other commenters here, https://github.com/CliMA/Oceananigans.jl immediately comes to mind.
> JAX offers more than just a JIT compiler: JAX functions are also differentiable
if the downstream library is completely implemented in JAX (numba) ecosystem. Similar for Julia, except implementing fast code in Julia is natural, but many python library is only differentiable because the 100x more effort in writing C/C++ backend, binding to python, and writing chain rules for foreign functions.
I would imagine Julia to be a good fit for this direction in the future!
True, but unfortunately Pytorch is not quite there yet when it comes to more complex benchmarks:
https://github.com/dionhaefner/pyhpc-benchmarks#example-resu...
JAX really is the only library that comes close to low-level code on CPU, almost always.
https://github.com/FluxML/XLA.jl
When in doubt, piggybacking on (or at least interoperating with) what the large technology companies are investing in is probably savvy, sort of what the OP did.