Senior Machine Learning Engineer

Cranium AI, Inc. 

Position:  Senior Machine Learning Engineer

Responsible for developing state-of-the-art AI based solutions for Cranium. Demonstrate deep expertise in graph-based modeling and analysis techniques, including community detection, and graph neural networks, to unlock rich insights from complex, interconnected data. Leverage advanced natural language processing capabilities, such as named entity recognition, sentiment analysis, and transformer-based language models, including large language models (LLMs) and generative AI (GenAI) techniques, to extract valuable information from unstructured data sources. Contribute innovative reinforcement learning solutions, leveraging cloud-based platforms to power multi-agent simulations and decision-making systems. Serve as a technical mentor, guiding junior data scientists and developers on cloud infrastructure, data pipeline design, and dashboard development. May telecommute from any location in the U.S.

This position requires a Master’s degree or equivalent in Computer Science, Computer Engineering, or a related field and 6 years related software development experience. Must also have demonstrated experience with each of the following: 1) Graph-based collaborative filtering, community detection algorithms, PageRank, and centrality measures for topic modeling using Neo4j or GraphX and proficiency in graph neural networks (GNNs) and heterogeneous graph analysis for enriched insights and predictions; 2) Using advanced NLP techniques including named entity recognition, part-of-speech tagging, and dependency parsing with spaCy, Stanford NLP, or NLTK, BERT model optimization using Hugging Face Transformers, advanced large language models (LLMs) like GPT, T5, Llama, and dimensionality reduction with PCA and t-SNE, alongside sentence embeddings via BERT or Universal Sentence Encoder; 3) Reinforcement learning algorithms such as DQN, PPO, and Actor-Critic Methods using OpenAI Gym and distributing it across using Ray RLlib for multi-agent simulations with SUMO and integration with cloud-based platforms like AWS SageMaker, GCP Vertex AI, or Azure AI studio; 4)  Mentoring junior data scientists and developers to develop the cloud infrastructure on AWS and backend dashboards; and 5) Designing distributed architecture for data ingestion implementing end-to-end ETL and ELT pipelines using job orchestrators and frameworks like Kafka, Airflow, and Spark Structured Streaming with Kubernetes, Docker and Terraform across multi-cloud environments (AWS, Azure, GCP) with experience in data warehousing, data lakes, and cloud-native data processing. 10% domestic travel required for company meetings. Employer will accept experience gained before, during, or after Master’s program. Employer will accept experience gained concurrently. Telecommuting available from anywhere in US. HQ at 1200 Morris Turnpike, Suite 3005, Short Hills, NJ 07078. Full-time. Salary: $169,541 to $180,000.  Please apply online at https://cranium.ai/company/careers/.