Skip to content

Ecosystem

Frameworks Extending Lux.jl

DiffEqFlux.jl

DiffEqFlux.jl

Universal neural differential equations with O(1) backprop, GPUs, and stiff+non-stiff DE solvers, demonstrating scientific machine learning (SciML) and physics-informed machine learning methods

SciMLSensitivity.jl

SciMLSensitivity.jl

A component of the DiffEq ecosystem for enabling sensitivity analysis for scientific machine learning (SciML). Optimize-then-discretize, discretize-then-optimize, adjoint methods, and more for ODEs, SDEs, DDEs, DAEs, etc.

NeuralPDE.jl

NeuralPDE.jl

Physics-Informed Neural Networks (PINN) and Deep BSDE Solvers of Differential Equations for Scientific Machine Learning (SciML) accelerated simulation

NeuralLyapunov.jl

NeuralLyapunov.jl

A library for searching for neural Lyapunov functions in Julia

DeepEquilibriumNetworks.jl

DeepEquilibriumNetworks.jl

Implicit Layer Machine Learning via Deep Equilibrium Networks, O(1) backpropagation with accelerated convergence

AbstractCosmologicalEmulators.jl

AbstractCosmologicalEmulators.jl

Repository containing the abstract interface to the emulators used in the CosmologicalEmulators organization

ContinuousNormalizingFlows.jl

ContinuousNormalizingFlows.jl

Implementations of Infinitesimal Continuous Normalizing Flows Algorithms in Julia

Sophon.jl

Sophon.jl

Efficient, Accurate, and Streamlined Training of Physics-Informed Neural Networks

DataDrivenDiffEq.jl

DataDrivenDiffEq.jl

Data driven modeling and automated discovery of dynamical systems for the SciML Scientific Machine Learning organization

NeuralGraphPDE.jl

NeuralGraphPDE.jl

Integrating Neural Ordinary Differential Equations, the Method of Lines, and Graph Neural Networks

Solaris.jl

Solaris.jl

Lightweight module for fusing physical and neural models

Boltz.jl

Boltz.jl

Accelerate your ML research using pre-built Deep Learning Models with Lux

GeometricMachineLearning.jl

GeometricMachineLearning.jl

Structure Preserving Machine Learning Models in Julia

Automatic Differentiation

Zygote.jl

Zygote.jl

Lux.jl default choice for AD

Tracker.jl

Tracker.jl

Well tested and robust AD library (might fail on edge cases)

ForwardDiff.jl

ForwardDiff.jl

For forward mode AD support

ReverseDiff.jl

ReverseDiff.jl

Tape based reverse mode AD (might fail on edge cases and doesn't work on GPU)

Enzyme.jl

Enzyme.jl

Experimental Support but will become the Future Default

Data Manipulation, Data Loading & Datasets

Augmentor.jl

Augmentor.jl

Data augmentation for machine learning

MLUtils.jl

MLUtils.jl

Utilities and abstractions for Machine Learning tasks

MLDatasets.jl

MLDatasets.jl

Utility package for accessing common Machine Learning datasets in Julia

Images.jl

Images.jl

An image library for Julia

DataAugmentation.jl

DataAugmentation.jl

Flexible data augmentation library for machine and deep learning

Neural Network Primitives

NNlib.jl

NNlib.jl

Neural Network primitives with multiple backends

LuxLib.jl

LuxLib.jl

Backend for Lux.jl

Optimization

Optimization.jl

Optimization.jl

Unified API for Optimization in Julia

Optimisers.jl

Optimisers.jl

Optimisers.jl defines many standard optimisers and utilities for learning loops

ParameterSchedulers.jl

ParameterSchedulers.jl

Common hyperparameter scheduling for ML

Parameter Manipulation

Functors.jl

Functors.jl

Parameterise all the things

ComponentArrays.jl

ComponentArrays.jl

Arrays with arbitrarily nested named components

Serialization

Serialization.jl

Serialization.jl

Provides serialization of Julia objects

JLD2.jl

JLD2.jl

HDF5-compatible file format in pure Julia

Testing Utilities

FiniteDiff.jl

FiniteDiff.jl

Fast non-allocating calculations of gradients, Jacobians, and Hessians with sparsity support

FiniteDifferences.jl

FiniteDifferences.jl

High accuracy derivatives, estimated via numerical finite differences (formerly FDM.jl)

JET.jl

JET.jl

JET employs Julia's type inference system to detect potential bugs and type instabilities

LuxTestUtils.jl

LuxTestUtils.jl

Collection of Functions useful for testing various packages in the Lux Ecosystem

Training Visualization & Logging

MLFlowClient.jl

MLFlowClient.jl

Julia client for MLFlow

TensorBoardLogger.jl

TensorBoardLogger.jl

Easy peasy logging to TensorBoard with Julia

Wandb.jl

Wandb.jl

Unofficial Julia bindings for logging experiments to wandb.ai