[D] Any less-boilerplate framework for Jax/Flax/Haiku?

This page summarizes the projects mentioned and recommended in the original post on /r/MachineLearning
Deep Learning Jax Awesome awesome-list neural-networks

InfluxDB - Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
featured
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com
featured

  • awesome-jax

    JAX - A curated list of resources https://github.com/google/jax

    • I have been looking for a framework/library that can be like/smiliar to PyTorch Lightning. I even checked on Awesome-Jax. Do we have any framework but for Jax/flax/haiku?

  • elegy

    A High Level API for Deep Learning in JAX

    • Elegy might be worth a look.

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo

  • equinox

    Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

    • GitHub: https://github.com/patrick-kidger/equinox

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts

  • 24 GitHub repos with 372M views that you can't miss out as a software engineer

    4 projects | dev.to | 25 Jan 2024

  • Awesome-Quant-Machine-Learning-Trading: curated list of books/online courses/youtube videos/blogs/interviews/papers/code etc. Updates are pretty infrequent Other Models - star count:2064.0

    1 project | /r/algoprojects | 17 Oct 2023

  • [P] Optimistix, nonlinear optimisation in JAX+Equinox!

    3 projects | /r/MachineLearning | 14 Oct 2023

  • Equinox: Elegant easy-to-use neural networks in Jax

    1 project | news.ycombinator.com | 18 Sep 2023

  • Show HN: Equinox (1.3k stars), a JAX library for neural networks and sciML

    1 project | news.ycombinator.com | 5 Sep 2023