Lightning fast C++/CUDA neural network framework
Sandbox for training deep learning networks
Scalable and user friendly neural forecasting algorithms.
A Julia framework for invertible neural networks
Message Passing Neural Networks for Molecule Property Prediction
Physics-Informed Neural Networks (PINN) Solvers
Internet-scale Neural Networks
BitNet: Scaling 1-bit Transformers for Large Language Models
High-performance neural network inference framework for mobile
DNN && GAN && NLP && BIG DATA
Open source machine learning framework
Pre-built implicit layer architectures with O(1) backprop, GPUs
Python-based neural networks API
SpikingJelly is an open-source deep learning framework
Simulation of spiking neural networks (SNNs) using PyTorch
Training neural networks on Apple Neural Engine via APIs
A JAX research toolkit to build, edit, & visualize neural networks
Deep learning driven jazz generation using Keras & Theano
DepGraph: Towards Any Structural Pruning
CV, NLP, LLM project applications, and advanced engineering deployment
Neural Network primitives with multiple backends
Handwritten Text Recognition (HTR) system implemented with TensorFlow
Global weather forecasting model using graph neural networks and JAX
Pre-trained Neural Network models in Axon