tensorflow/tensorflowGitHub
calls dm-haiku “JAX-based network modules; choose if you're already on JAX and want a lightweight TF-free alternative to tf.Module.”
An Open Source Machine Learning Framework for Everyone
198.0kC++active
google-deepmind/dm-haiku Alternatives
dm-haiku is queued for analysis. Crawling candidates and ranking them takes a few minutes. This page updates itself when the list exists — you can also come back later.
What is known so far comes from the other direction:
Listed as an alternative to
These projects were analysed and named dm-haiku among their alternatives. The relationship is not symmetric — how dm-haiku rates them is a separate judgement, made when dm-haiku is analysed in its own right.
pytorch/pytorchGitHub
calls dm-haiku “choose if you build on JAX and want a Sonnet-like neural network module library.”
Tensors and Dynamic neural networks in Python with strong GPU acceleration
102.7kPythonactive
keras-team/kerasGitHub
calls dm-haiku “Choose when working directly in JAX and wanting a light, functional NN library instead of Keras's full API.”
Deep Learning for humans
64.3kPythonactive
tensorflow/tfjsGitHub
calls dm-haiku “when you use JAX and need a neural network library with a familiar object-oriented API.”
A WebGL accelerated JavaScript library for training and deploying ML models.
19.1kTypeScriptactive
tracel-ai/burnGitHub
calls dm-haiku “Choose when you are building on JAX and want a small, functional NN library instead of a full framework.”
Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
15.8kRustactive
Oneflow-Inc/oneflowGitHub
calls dm-haiku “choose when you already use JAX and want a simple neural network module library”
OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.
9.4kC++slowing
apple/corenetGitHub
calls dm-haiku “When you're already on JAX and only need lightweight NN module primitives.”
CoreNet: A library for training deep neural networks
7.0kJupyter Notebookslowing