1 repo
Awesome GitHub RepositoriesGradient Computation Engines
Frameworks for calculating precise gradients and derivatives to support model optimization.
Distinguishing note: Specifically targets the workflow of gradient calculation for optimization, distinct from general model training libraries.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Gradient Computation Engines. Refine with filters or upvote what's useful.
Awesome Gradient Computation Engines GitHub Repositories
Find the best repos with AI.We'll search the best matching repositories with AI.
- jax-ml/jax
jax-ml/jax
34,919View on GitHubThis project is a high-performance numerical computing library designed for large-scale scientific and machine learning workloads. It functions as an automatic differentiation framework and a just-in-time compilation engine, transforming high-level Python code into optimized machine instructions. By enforcing pure functional programming patterns and immutable array semantics, the library ensures that mathematical functions remain compatible with automated graph transformations and symbolic differentiation. The platform distinguishes itself through its distributed array computing capabilities,
Enables efficient optimization and training of complex systems by calculating precise gradients and derivatives.
Pythonjax
