A general-purpose probabilistic programming system
Scalable and user friendly neural forecasting algorithms.
Pre-trained Neural Network models in Axon
DeepONets, Neural Operators, Physics-Informed Neural Ops in Julia
Lightning fast C++/CUDA neural network framework
A JAX research toolkit to build, edit, & visualize neural networks
High-performance neural network inference framework for mobile
Instant neural graphics primitives: lightning fast NeRF and more
oneAPI Deep Neural Network Library (oneDNN)
Visualizer for neural network, deep learning, machine learning models
Predicting stock prices using a TensorFlow LSTM
High-efficiency floating-point neural network inference operators
BitNet: Scaling 1-bit Transformers for Large Language Models
Relax! Flux is the ML library that doesn't make you tensor
Global weather forecasting model using graph neural networks and JAX
CoreNet: A library for training deep neural networks
JAX-based neural network library
TensorFlow.js is a library for machine learning in JavaScript
Neural Network primitives with multiple backends
Simulation of spiking neural networks (SNNs) using PyTorch
Hummingbird compiles trained ML models into tensor computation
Stanford NLP Python library for many human languages
Hundreds of fully solved job interview questions
PRML algorithms implemented in Python
Sandbox for training deep learning networks