AI Packages

Flux.jl4466Relax! Flux is the ML library that doesn't make you tensor

MLJ.jl1779A Julia machine learning framework

Knet.jl1427Koç University deep learning framework.

Mocha.jl1287Deep Learning framework for Julia

BrainFlow.jl1273BrainFlow is a library intended to obtain, parse and analyze EEG, EMG, ECG and other kinds of data from biosensors

AlphaZero.jl1232A generic, simple and fast implementation of Deepmind's AlphaZero algorithm.

NeuralNetDiffEq.jl966PhysicsInformed Neural Networks (PINN) Solvers of (Partial) Differential Equations for Scientific Machine Learning (SciML) accelerated simulation

TensorFlow.jl884A Julia wrapper for TensorFlow

DiffEqFlux.jl861Prebuilt implicit layer architectures with O(1) backprop, GPUs, and stiff+nonstiff DE solvers, demonstrating scientific machine learning (SciML) and physicsinformed machine learning methods

FastAI.jl589Repository of best practices for deep learning in Julia, inspired by fastai

ReinforcementLearning.jl583A reinforcement learning package for Julia

ScikitLearn.jl546Julia implementation of the scikitlearn API https://cstjean.github.io/ScikitLearn.jl/dev/

Transformers.jl521Julia Implementation of Transformer models

Lux.jl479Elegant & Performant Scientific Machine Learning in Julia

Enzyme.jl438Julia bindings for the Enzyme automatic differentiator

StatisticalRethinking.jl386Julia package with selected functions in the R package `rethinking`. Used in the SR2... projects.

TextAnalysis.jl373Julia package for text analysis

MXNet.jl371MXNet Julia Package  flexible and efficient deep learning in Julia

AutoMLPipeline.jl355A package that makes it trivial to create and evaluate machine learning pipeline architectures.

Clustering.jl353A Julia package for data clustering

DecisionTree.jl351Julia implementation of Decision Tree (CART) and Random Forest algorithms

Metalhead.jl328Computer vision models for Flux

Dojo.jl307A differentiable physics engine for robotics

SimpleChains.jl234Simple chains

MLDatasets.jl227Utility package for accessing common Machine Learning datasets in Julia

GraphNeuralNetworks.jl218Graph Neural Networks in Julia

AbstractGPs.jl217Abstract types and methods for Gaussian Processes.

Torch.jl211Sensible extensions for exposing torch in Julia.

ReservoirComputing.jl206Reservoir computing utilities for scientific machine learning (SciML)

Reinforce.jl201Abstractions, algorithms, and utilities for reinforcement learning in Julia

NNlib.jl201Neural Network primitives with multiple backends

MLBase.jl186A set of functions to support the development of machine learning algorithms

EvoTrees.jl175Boosted trees in Julia

AdaGram.jl170Adaptive Skipgram implementation in Julia

SeaPearl.jl168Julia hybrid constraint programming solver enhanced by a reinforcement learning driven search.

MLJBase.jl160Core functionality for the MLJ machine learning framework

Yota.jl158Reversemode automatic differentiation in Julia

ForneyLab.jl149Julia package for automatically generating Bayesian inference algorithms through message passing on Forneystyle factor graphs.

InvertibleNetworks.jl149A Julia framework for invertible neural networks

LossFunctions.jl147Julia package of loss functions for machine learning.
Loading more...