google/flax
quality grade A, 92 out of 100Flax is a neural network library for JAX that is designed for flexibility.
- stars
- 7.4k
- stars gained this week
- +4this week
- forks, open issues and contributors
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Core deep-learning frameworks and libraries for pretraining and distributed training.
Signals: deep-learning, neural-network, pytorch, tensorflow, jax, distributed-training, training, deepspeed
2,663 results
Flax is a neural network library for JAX that is designed for flexibility.
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Formula Recognition & Office Editing | Handwriting & PDF to LaTeX/Markdown | Secure API Integrations.
Open standard for machine learning interoperability
Stock AI Trader: 1-stop platform for learning, sim & live trading. Covers: stock basics, strategies, LLMs, factor mining, ML/DL/RL, graph nets, HFT, C++ deploy & JoinQuant code. 股票AI操盘手:一站式学习、模拟、实盘平台。涵盖:股票基础、策略、大模型、因子挖掘、机器学习/深度学习/强化学习、图网络、高频交易、C++部署及聚宽代码。
An LLM post-training framework with vLLM for RL Scaling
The ultimate training toolkit for finetuning diffusion models
ncnn is a high-performance neural network inference framework optimized for the mobile platform
Comprehensive optical design, optimization, and analysis in Python, including GPU-accelerated and differentiable ray tracing via PyTorch.
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.
Machine Learning Engineering Open Book
XR Blocks is a lightweight WebXR + AI library for rapidly prototyping advanced AI + XR experiences.
TensorFlow plugin for the Unreal Engine.
Open-source library of optimized deep learning operations (matmul, convolution, attention) for CPUs (x64, AArch64, RISC-V) and Intel GPUs. Used by PyTorch, TensorFlow, OpenVINO, and ONNX Runtime.
Distributed GPU-Accelerated Framework for Evolutionary Computation. Comprehensive Library of Evolutionary Algorithms & Benchmark Problems.
DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning
ModelScope: bring the notion of Model-as-a-Service to life.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
The official CLI and Python client for the Hugging Face Hub.
Low-code framework for building custom LLMs, neural networks, and other AI models
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
A deep learning toolkit for healthcare AI: EHR (MIMIC, eICU, OMOP), clinical text, signals and imaging. 30+ models, reproducible pipelines.
On-device AI across mobile, embedded and edge for PyTorch
An Engine-Agnostic Deep Learning Framework in Java
24,545 repositories in the index in total.