Search Results for "automatic differentiation"
Sort By:
Showing 25 open source projects for "automatic differentiation"
View related business solutions-
$300 Free Credits to Build on Google CloudPut your $300 in credit toward real workloads, then keep building with free monthly usage for 20+ products. No commitment and no charge until you upgrade.
-
Ship Agents FasterGemini Enterprise Agent Platform lets you rapidly build, scale, govern and optimize production-ready agents grounded in your organization's data. The platform enables developers to build custom or pre-built agents for virtually any use case. New customers get $300 in free credits.
-
1
Diffrax
Numerical differential equation solvers in JAX
...The project provides tools for solving ordinary differential equations, stochastic differential equations, controlled differential equations, and related systems in a way that fits naturally into modern machine learning and differentiable programming workflows. Because it is written to work closely with JAX, it supports just-in-time compilation, automatic differentiation, vectorization, and accelerator-backed execution on hardware such as GPUs and TPUs. This makes it especially appealing for researchers who need equation solvers that can be embedded inside trainable models or simulation-heavy learning systems. -
2
micrograd
A tiny scalar-valued autograd engine and a neural net library
micrograd is a tiny, educational automatic differentiation engine focused on scalar values, built to show how backpropagation works end to end with minimal code. It constructs a dynamic computation graph as you perform math operations and then computes gradients by walking that graph backward, making it an approachable “from scratch” autograd reference. On top of the core autograd “Value” concept, the project includes a small neural network library that lets you define and train simple models with a PyTorch-like feel, including multilayer perceptrons. ... -
3
TensorFlow Quantum
Open-source Python framework for hybrid quantum-classical ml learning
...TensorFlow Quantum integrates with the Cirq quantum computing framework to define and manipulate quantum circuits, while leveraging TensorFlow’s infrastructure for optimization, automatic differentiation, and large-scale computation. The library also supports high-performance simulation of quantum circuits, enabling researchers to test and evaluate quantum models even without direct access to quantum hardware. -
4
JAX
Composable transformations of Python+NumPy programs
...But JAX also lets you just-in-time compile your own Python functions into XLA-optimized kernels using a one-function API, jit. Compilation and automatic differentiation can be composed arbitrarily, so you can express sophisticated algorithms and get maximal performance without leaving Python. -
Custom VMs From 1 to 96 vCPUs With 99.95% UptimeLive migration and automatic failover keep workloads online through maintenance. One free e2-micro VM every month.
-
5
Tequila
A High-Level Abstraction Framework for Quantum Algorithms
Tequila is an abstraction framework for (variational) quantum algorithms. It operates on abstract data structures allowing the formulation, combination, automatic differentiation and optimization of generalized objectives. Tequila can execute the underlying quantum expectation values on state-of-the-art simulators as well as on real quantum devices. -
6
PennyLane
A cross-platform Python library for differentiable programming
A cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network. Built-in automatic differentiation of quantum circuits, using the near-term quantum devices directly. You can combine multiple quantum devices with classical processing arbitrarily! Support for hybrid quantum and classical models, and compatible with existing machine learning libraries. Quantum circuits can be set up to interface with either NumPy, PyTorch, JAX, or TensorFlow, allowing hybrid CPU-GPU-QPU computations. ... -
7
Haiku Sonnet for JAX
JAX-based neural network library
Haiku is a library built on top of JAX designed to provide simple, composable abstractions for machine learning research. JAX is a numerical computing library that combines NumPy, automatic differentiation, and first-class GPU/TPU support. Haiku is a simple neural network library for JAX that enables users to use familiar object-oriented programming models while allowing full access to JAX's pure function transformations. Haiku provides two core tools: a module abstraction, hk.Module, and a simple function transformation, hk.transform. hk.Modules are Python objects that hold references to their own parameters, other modules, and methods that apply functions on user inputs. hk.transform turns functions that use these object-oriented, functionally "impure" modules into pure functions that can be used with jax.jit, jax.grad, jax.pmap, etc. -
8
neural-style in TensorFlow
Neural style in TensorFlow
neural-style in TensorFlow is a TensorFlow implementation of neural style transfer. It creates a new image by combining the content of one image with the artistic style of one or more style images. The project uses TensorFlow automatic differentiation and the Adam optimizer rather than the original L-BFGS approach. Users can run it from the command line by providing a content image, style image inputs, and an output path. It supports checkpoint outputs, iteration control, style blending, and hyperparameter tuning for content weight, style weight, and learning rate. ... -
9
VibeTensor
Our first fully AI generated deep learning system
VibeTensor is a groundbreaking open-source research system software stack for deep learning that was uniquely generated almost entirely by AI coding agents under guided human supervision, demonstrating a new frontier in AI-assisted software engineering. It implements a PyTorch-style eager tensor library with a modern C++20 core that supports both CPU and CUDA backends, giving it the ability to manage tensors, automatic differentiation (autograd), and complex computation flows similar to mainstream frameworks. What makes VibeTensor remarkable is that every major component, from core libraries and dispatch systems to CUDA runtime support, caching allocators, and language bindings, was created and validated by coding agents using automated builds and tests rather than manual line-by-line human coding. ... -
MongoDB Atlas runs apps anywhereMongoDB Atlas gives you the freedom to build and run modern applications anywhere—across AWS, Azure, and Google Cloud. With global availability in over 115 regions, Atlas lets you deploy close to your users, meet compliance needs, and scale with confidence across any geography.
-
10
RLax
Library of JAX-based building blocks for reinforcement learning agents
RLax (pronounced “relax”) is a JAX-based library developed by Google DeepMind that provides reusable mathematical building blocks for constructing reinforcement learning (RL) agents. Rather than implementing full algorithms, RLax focuses on the core functional operations that underpin RL methods—such as computing value functions, returns, policy gradients, and loss terms—allowing researchers to flexibly assemble their own agents. It supports both on-policy and off-policy learning, as well as... -
11
Amazon Braket PennyLane Plugin
A plugin for allowing Xanadu PennyLane to use Amazon Braket devices
...The Amazon Braket Python SDK is an open-source library that provides a framework to interact with quantum computing hardware devices and simulators through Amazon Braket. PennyLane is a machine learning library for optimization and automatic differentiation of hybrid quantum-classical computations. Once the Pennylane-Braket plugin is installed, the provided Braket devices can be accessed straight away in PennyLane, without the need to import any additional packages. While the local device helps with small-scale simulations and rapid prototyping, the remote device allows you to run larger simulations or access quantum hardware via the Amazon Braket service. -
12
Shumai
Fast Differentiable Tensor Library in JavaScript & TypeScript with Bun
...It provides a high-performance framework for numerical computing and machine learning within modern JavaScript runtimes. Built on Bun and Flashlight, with ArrayFire as its numerical backend, Shumai brings GPU-accelerated tensor operations, automatic differentiation, and scientific computing tools directly to JavaScript developers. It allows seamless integration of machine learning, deep learning, and custom differentiable programs into web-based or server-side environments without relying on Python frameworks. The library supports matrix operations, gradient computation, and tensor conversions with intuitive APIs and near-native speed, thanks to Bun’s low-overhead FFI bindings. ... -
13
snntorch
Deep and online learning with spiking neural networks in Python
...The library extends PyTorch’s tensor computation capabilities to support gradient-based learning for networks composed of spiking neurons. This allows researchers to train spiking neural models using familiar deep learning workflows while taking advantage of GPU acceleration and automatic differentiation. snnTorch provides implementations of common spiking neuron models, surrogate gradient training methods, and utilities for handling temporal neural dynamics. Because spiking neural networks operate over time and encode information through spike timing, the library includes tools for simulating temporal behavior. -
14
LLM Training Puzzles
What would you do with 1000 H100s
...The hands-on format emphasizes deriving behavior rather than memorizing framework commands. Google Colab is recommended so readers can copy the notebook and begin experimenting quickly. The project belongs to a broader puzzle series covering tensors, automatic differentiation, Transformers, GPUs, and related topics. -
15
Paddle Quantum
Paddle Quantum
Paddle Quantum (量桨) is the world's first cloud-integrated quantum machine learning platform based on Baidu PaddlePaddle. It supports the building and training of quantum neural networks, making PaddlePaddle the first deep-learning framework in China. Paddle Quantum is feature-rich and easy to use. It provides comprehensive API documentation and tutorials help users get started right away. Paddle Quantum aims at establishing a bridge between artificial intelligence (AI) and quantum computing... -
16
eat TensorFlow2 in 30 days
Educational project that teaches developers how to train models
...It organizes learning materials into a structured curriculum designed to progressively introduce essential TensorFlow concepts and APIs. The tutorials cover tensors, automatic differentiation, computational operations, and neural network development. Users learn how to construct models with Keras and implement custom training workflows using lower-level TensorFlow APIs. The project also explains model evaluation, optimization, callbacks, and training with GPU and TPU hardware. Advanced lessons explore model deployment through TensorFlow Serving and integration with external systems. ... -
17
Chainer
A flexible deep learning framework
Chainer is a Python-based deep learning framework. It provides automatic differentiation APIs based on dynamic computational graphs as well as high-level APIs for neural networks. -
18
Arraymancer
A fast, ergonomic and portable tensor library in Nim
Arraymancer is a tensor and deep learning library for the Nim programming language, designed for high-performance numerical computations and machine learning applications. -
19
gradslam
gradslam is an open source differentiable dense SLAM library
...However, learning representations for SLAM has been an open question, because traditional SLAM systems are not end-to-end differentiable. In this work, we present gradSLAM, a differentiable computational graph take on SLAM. Leveraging the automatic differentiation capabilities of computational graphs, gradSLAM enables the design of SLAM systems that allow for gradient-based learning across each of their components, or the system as a whole. -
20
CrypTen
A framework for Privacy Preserving Machine Learning
...Designed to make secure computation accessible to machine learning practitioners, CrypTen introduces a CrypTensor object that behaves like a regular PyTorch tensor, allowing users to seamlessly apply automatic differentiation and neural network operations. Its design mirrors PyTorch’s modular and library-based structure, enabling flexible experimentation, debugging, and model development. The framework supports both encryption and decryption of tensors and operations such as addition and multiplication over encrypted values. Although not yet production-ready, CrypTen focuses on advancing real-world secure ML applications, such as training and inference over private datasets, without exposing sensitive data. -
21
TF Quant Finance
High-performance TensorFlow library for quantitative finance
...It implements pricing engines, risk measures, stochastic models, optimizers, and random number generators that are differentiable and vectorized for accelerators. Users can value options and fixed-income instruments, simulate paths, fit curves, and calibrate models while leveraging TensorFlow’s jit compilation and automatic differentiation. The codebase is organized as modular math and finance primitives so you can combine building blocks or target end-to-end examples. It includes Bazel builds, tests, and example notebooks to accelerate learning and adoption in real workflows. With hardware acceleration and differentiable models, it enables modern techniques like gradient-based calibration and end-to-end learning of market dynamics. -
22
A symbolic framework for C++, Python and Octave implementing automatic differentiation by source code transformation in forward and reverse modes on sparse matrix-valued computational graphs.
-
23
Tangent
Source-to-source debuggable derivatives in pure Python
Existing libraries implement automatic differentiation by tracing a program's execution (at runtime, like PyTorch) or by staging out a dynamic data-flow graph and then differentiating the graph (ahead-of-time, like TensorFlow). In contrast, Tangent performs ahead-of-time autodiff on the Python source code itself, and produces Python source code as its output. Tangent fills a unique location in the space of machine learning tools. -
24
autodiff
Python class for automatic differentiation
Python automatic differentiation class for forward mode automatic differentiation using operator overloading and reimplemented math functions. Single and partial derivatives are supported. Supported operators: +, -, *, /, **, +=, -=, *=, /=, **= Available functions: sin, cos, tan, asin, acos, atan, sqrt, exp, log, log10, sinh, cosh, tanh, asinh, acosh, atanh See README file for usage examples. -
25
- Previous
- You're on page 1
- Next
